pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Datasets, Transforms and Models specific to Computer Vision

PyTorch tutorials.

A PyTorch native platform for training generative AI models

On-device AI across mobile, embedded and edge for PyTorch

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Serve, optimize and scale PyTorch models in production

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

Run PyTorch LLMs locally on servers, desktop and mobile

Models, data loaders and abstractions for language processing, powered by PyTorch

ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation

Compiler for Neural Network hardware accelerators

PyTorch native quantization for training and inference

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Data manipulation and transformation for audio signal processing, powered by PyTorch

Enabling PyTorch on XLA Devices (e.g. Google TPU)

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

PyTorch android examples of usage in applications

Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators

Collective communications library with various primitives for multi-machine training.

Submission to https://pytorch.org/hub/

functorch is JAX-like composable function transforms for PyTorch.

C++ extensions in PyTorch

CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

A dict[str, torch.Tensor] that behaves like one tensor, and TensorClass: typed, batched dataclasses that compile.

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

Translate - a PyTorch Language Library

Pipeline Parallelism for PyTorch

PyTorch elastic training

High performance model preprocessing library on PyTorch

PyTorch iOS examples

TVM integration into PyTorch

Implementations of ideas from recent papers

Accelerate PyTorch models with ONNX Runtime

Continuous builder and binary build scripts for pytorch

high performance image loading and augmenting routines mimicking PIL.Image interface

The website for PyTorch

A tensor-aware point-to-point communication primitive for machine learning

PyTorch C++ API Documentation

Examples of C extensions for PyTorch

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

PyTorch RFCs (experimental)

Torch Distributed Experimental

This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our...

Example repository for custom C++/CUDA operators for TorchScript

PyTorch Sphinx Theme