Discover pytorch/dr-ci Open Source project by @pytorch

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

83,553

examples

Datasets, Transforms and Models specific to Computer Vision

22,311

vision

tutorials

captum

Model interpretability and understanding for PyTorch

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

4,781

ignite

Serve, optimize and scale PyTorch models in production

4,507

serve

Java

4,190

torchtune

PyTorch native finetuning library

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

4,163

text

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

3,490

ELF

Compiler for Neural Network hardware accelerators

3,364

glow

Bayesian optimization in PyTorch

3,197

botorch

Run PyTorch LLMs locally on servers, desktop and mobile

3,043

torchchat

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

3,040

TensorRT

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

2,565

audio

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

2,471

xla

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

2,469

A native PyTorch Library for large model training

2,241

torchtitan

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

2,130

executorch

Pytorch domain library for recommendation systems

1,954

torchrec

Training PyTorch models with differential privacy

1,852

opacus

A lightweight library for PyTorch training tools and utilities

1,666

tnt

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

1,650

QNNPACK

1,519

android-demo-app

PyTorch android examples of usage in applications

Java

1,460

functorch

functorch is JAX-like composable function transforms for PyTorch.

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

1,388

hub

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

1,384

FBGEMM

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

1,156

data

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

1,112

cpuinfo

989

torchdynamo

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

C++ extensions in PyTorch

989

extension-cpp

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

980

benchmark

Translate - a PyTorch Language Library

841

translate

TensorDict is a pytorch dedicated tensor container.

820

tensordict

816

elastic

PyTorch elastic training

Pipeline Parallelism for PyTorch

728

PiPPy

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

698

kineto

High performance model preprocessing library on PyTorch

682

torcharrow

PyTorch native quantization and sparsity for training and inference

641

ios-demo-app

tvm

TVM integration into PyTorch

Implementations of ideas from recent papers

451

contrib

Accelerate PyTorch models with ONNX Runtime

390

ort

Continuous builder and binary build scripts for pytorch

353

builder

Shell

325

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.

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

319

accimage

317

extension-ffi

Examples of C extensions for PyTorch

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

257

nestedtensor

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

252

tensorpipe

247

pytorch.github.io

The website for PyTorch

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.

222

torcheval

PyTorch C++ API Documentation

210

cppdocs

This is a repository for all workshop related materials.

206

workshops

Configuration classes enabling type-safe PyTorch configuration for Hydra apps

204

hydra-torch

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.

199

multipy

A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.

169

torchsnapshot

java-demo

rfcs

PyTorch RFCs (experimental)

120

torchdistx

Torch Distributed Experimental

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

115

extension-script

Cryptographically secure pseudorandom number generators for PyTorch

112

csprng

pytorch_sphinx_theme

test-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.

TypeScript

expecttest

torchcodec

PyTorch video decoding

MaskedTensors for PyTorch

maskedtensor

A GitHub action to run clang-tidy and annotate failures

add-annotations-github-action

JavaScript

ci-hud

HUD for CI activity on `pytorch/pytorch`, provides a top level view for jobs to easily discern regressions

JavaScript

probot

PyTorch GitHub bot written in probot

TypeScript

ossci-job-dsl

Jenkins job definitions for OSSCI

Groovy

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

Makefile

docs

This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.