Pinned Repositories
captum
Model interpretability and understanding for PyTorch
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
serve
Serve, optimize and scale PyTorch models in production
text
Models, data loaders and abstractions for language processing, powered by PyTorch
torchrec
Pytorch domain library for recommendation systems
torchtune
A Native-PyTorch Library for LLM Fine-tuning
tutorials
PyTorch tutorials.
vision
Datasets, Transforms and Models specific to Computer Vision
pytorch's Repositories
pytorch/captum
Model interpretability and understanding for PyTorch
pytorch/ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
pytorch/opacus
Training PyTorch models with differential privacy
pytorch/android-demo-app
PyTorch android examples of usage in applications
pytorch/hub
Submission to https://pytorch.org/hub/
pytorch/torchdynamo
A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
pytorch/cpuinfo
CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
pytorch/extension-cpp
C++ extensions in PyTorch
pytorch/translate
Translate - a PyTorch Language Library
pytorch/elastic
PyTorch elastic training
pytorch/PiPPy
Pipeline Parallelism for PyTorch
pytorch/torcharrow
High performance model preprocessing library on PyTorch
pytorch/ios-demo-app
PyTorch iOS examples
pytorch/ort
Accelerate PyTorch models with ONNX Runtime
pytorch/torchx
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.
pytorch/extension-ffi
Examples of C extensions for PyTorch
pytorch/nestedtensor
[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
pytorch/tensorpipe
A tensor-aware point-to-point communication primitive for machine learning
pytorch/hydra-torch
Configuration classes enabling type-safe PyTorch configuration for Hydra apps
pytorch/workshops
This is a repository for all workshop related materials.
pytorch/torcheval
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.
pytorch/multipy
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.
pytorch/torchsnapshot
A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
pytorch/torchdistx
Torch Distributed Experimental
pytorch/rfcs
PyTorch RFCs (experimental)
pytorch/extension-script
Example repository for custom C++/CUDA operators for TorchScript
pytorch/csprng
Cryptographically secure pseudorandom number generators for PyTorch
pytorch/maskedtensor
MaskedTensors for PyTorch
pytorch/add-annotations-github-action
A GitHub action to run clang-tidy and annotate failures
pytorch/torchhub_testing
Repo to test torchhub. Nothing to see here.