There are no reviews yet. Be the first to send feedback to the community and the maintainers!
pytorch
Tensors and Dynamic neural networks in Python with strong GPU accelerationexamples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.vision
Datasets, Transforms and Models specific to Computer Visiontutorials
PyTorch tutorials.captum
Model interpretability and understanding for PyTorchignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.serve
Serve, optimize and scale PyTorch models in productiontorchtune
A Native-PyTorch Library for LLM Fine-tuningtext
Models, data loaders and abstractions for language processing, powered by PyTorchELF
ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementationglow
Compiler for Neural Network hardware acceleratorsbotorch
Bayesian optimization in PyTorchtorchchat
Run PyTorch LLMs locally on servers, desktop and mobileTensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRTaudio
Data manipulation and transformation for audio signal processing, powered by PyTorchxla
Enabling PyTorch on XLA Devices (e.g. Google TPU)rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.torchrec
Pytorch domain library for recommendation systemsopacus
Training PyTorch models with differential privacyexecutorch
On-device AI across mobile, embedded and edge for PyTorchtnt
A lightweight library for PyTorch training tools and utilitiesQNNPACK
Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operatorstorchtitan
A native PyTorch Library for large model trainingandroid-demo-app
PyTorch android examples of usage in applicationsfunctorch
functorch is JAX-like composable function transforms for PyTorch.hub
Submission to https://pytorch.org/hub/FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/data
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.cpuinfo
CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)torchdynamo
A Python-level JIT compiler designed to make unmodified PyTorch programs faster.extension-cpp
C++ extensions in PyTorchbenchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.translate
Translate - a PyTorch Language Librarytensordict
TensorDict is a pytorch dedicated tensor container.elastic
PyTorch elastic trainingPiPPy
Pipeline Parallelism for PyTorchkineto
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.torcharrow
High performance model preprocessing library on PyTorchao
PyTorch native quantization and sparsity for training and inferenceios-demo-app
PyTorch iOS examplestvm
TVM integration into PyTorchcontrib
Implementations of ideas from recent papersort
Accelerate PyTorch models with ONNX Runtimebuilder
Continuous builder and binary build scripts for pytorchtorchx
TorchX 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.accimage
high performance image loading and augmenting routines mimicking PIL.Image interfaceextension-ffi
Examples of C extensions for PyTorchnestedtensor
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.tensorpipe
A tensor-aware point-to-point communication primitive for machine learningpytorch.github.io
The website for PyTorchtorcheval
A 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.cppdocs
PyTorch C++ API Documentationworkshops
This is a repository for all workshop related materials.hydra-torch
Configuration classes enabling type-safe PyTorch configuration for Hydra appsmultipy
torch::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.torchsnapshot
A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.java-demo
rfcs
PyTorch RFCs (experimental)torchdistx
Torch Distributed Experimentalextension-script
Example repository for custom C++/CUDA operators for TorchScriptcsprng
Cryptographically secure pseudorandom number generators for PyTorchpytorch_sphinx_theme
PyTorch Sphinx Themetest-infra
This 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.expecttest
torchcodec
PyTorch video decodingmaskedtensor
MaskedTensors for PyTorchadd-annotations-github-action
A GitHub action to run clang-tidy and annotate failuresci-hud
HUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressionsprobot
PyTorch GitHub bot written in probotossci-job-dsl
Jenkins job definitions for OSSCIpytorch-integration-testing
Testing downstream libraries using pytorch release candidatestorchhub_testing
Repo to test torchhub. Nothing to see here.dr-ci
Diagnose and remediate CI jobspytorch-ci-dockerfiles
Scripts for generating docker images for PyTorch CIlabeler-github-action
GitHub action for labeling issues and pull requests based on conditionsLove Open Source and this site? Check out how you can help us