pytorchTensors and Dynamic neural networks in Python with strong GPU acceleration
examplesA set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
visionDatasets, Transforms and Models specific to Computer Vision
igniteHigh-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
captumModel interpretability and understanding for PyTorch
serveServe, optimize and scale PyTorch models in production
textModels, data loaders and abstractions for language processing, powered by PyTorch
ELFELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation
glowCompiler for Neural Network hardware accelerators
botorchBayesian optimization in PyTorch
audioData manipulation and transformation for audio signal processing, powered by PyTorch
xlaEnabling PyTorch on XLA Devices (e.g. Google TPU)
TensorRTPyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
tntA lightweight library for PyTorch training tools and utilities
QNNPACKQuantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators
torchrecPytorch domain library for recommendation systems
opacusTraining PyTorch models with differential privacy
functorchfunctorch is JAX-like composable function transforms for PyTorch.
hubSubmission to https://pytorch.org/hub/
rlA modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
android-demo-appPyTorch android examples of usage in applications
dataA PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
FBGEMMFB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
torchdynamoA Python-level JIT compiler designed to make unmodified PyTorch programs faster.
extension-cppC++ extensions in PyTorch
translateTranslate - a PyTorch Language Library
cpuinfoCPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
elasticPyTorch elastic training
benchmarkTorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
torcharrowHigh performance model preprocessing library on PyTorch
ios-demo-appPyTorch iOS examples
executorchEnd-to-end solution for enabling on-device AI across mobile and edge devices for PyTorch models
kinetoA CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
tvmTVM integration into PyTorch
contribImplementations of ideas from recent papers
ortAccelerate PyTorch models with ONNX Runtime
PiPPyPipeline Parallelism for PyTorch
accimagehigh performance image loading and augmenting routines mimicking PIL.Image interface
builderContinuous builder and binary build scripts for pytorch
extension-ffiExamples of C extensions for PyTorch
nestedtensor[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
tensorpipeA tensor-aware point-to-point communication primitive for machine learning
torchxTorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.
pytorch.github.ioThe website for PyTorch
hydra-torchConfiguration classes enabling type-safe PyTorch configuration for Hydra apps
cppdocsPyTorch C++ API Documentation
workshopsThis is a repository for all workshop related materials.
multipytorch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.
torchevalA library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.
extension-scriptExample repository for custom C++/CUDA operators for TorchScript
rfcsPyTorch RFCs (experimental)
torchsnapshotA performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
csprngCryptographically secure pseudorandom number generators for PyTorch
torchdistxTorch Distributed Experimental
pytorch_sphinx_themePyTorch Sphinx Theme
test-infraThis repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
maskedtensorMaskedTensors for PyTorch
add-annotations-github-actionA GitHub action to run clang-tidy and annotate failures
probotPyTorch GitHub bot written in probot
ossci-job-dslJenkins job definitions for OSSCI
ci-hudHUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressions
pytorch-integration-testingTesting downstream libraries using pytorch release candidates
torchhub_testingRepo to test torchhub. Nothing to see here.
dr-ciDiagnose and remediate CI jobs
pytorch-ci-dockerfilesScripts for generating docker images for PyTorch CI
labeler-github-actionGitHub action for labeling issues and pull requests based on conditions