@pytorch

Top repositories

1

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
Python
79,622
star
2

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python
21,905
star
3

vision

Datasets, Transforms and Models specific to Computer Vision
Python
15,622
star
4

tutorials

PyTorch tutorials.
Jupyter Notebook
7,906
star
5

captum

Model interpretability and understanding for PyTorch
Python
4,653
star
6

ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
Python
4,475
star
7

serve

Serve, optimize and scale PyTorch models in production
Java
4,029
star
8

text

Models, data loaders and abstractions for language processing, powered by PyTorch
Python
3,462
star
9

torchtune

A Native-PyTorch Library for LLM Fine-tuning
Python
3,385
star
10

ELF

ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation
C++
3,357
star
11

glow

Compiler for Neural Network hardware accelerators
C++
3,176
star
12

botorch

Bayesian optimization in PyTorch
Jupyter Notebook
2,980
star
13

audio

Data manipulation and transformation for audio signal processing, powered by PyTorch
Python
2,410
star
14

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Python
2,384
star
15

xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
C++
2,357
star
16

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Python
1,959
star
17

torchrec

Pytorch domain library for recommendation systems
Python
1,765
star
18

opacus

Training PyTorch models with differential privacy
Jupyter Notebook
1,639
star
19

tnt

A lightweight library for PyTorch training tools and utilities
Python
1,633
star
20

QNNPACK

Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators
C
1,513
star
21

android-demo-app

PyTorch android examples of usage in applications
Java
1,443
star
22

functorch

functorch is JAX-like composable function transforms for PyTorch.
Jupyter Notebook
1,377
star
23

executorch

On-device AI across mobile, embedded and edge for PyTorch
C++
1,373
star
24

hub

Submission to https://pytorch.org/hub/
Python
1,372
star
25

torchtitan

A native PyTorch Library for large model training
Python
1,198
star
26

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++
1,095
star
27

data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Python
1,075
star
28

torchdynamo

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

cpuinfo

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

extension-cpp

C++ extensions in PyTorch
Python
952
star
31

translate

Translate - a PyTorch Language Library
Python
816
star
32

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
Python
800
star
33

elastic

PyTorch elastic training
Python
728
star
34

PiPPy

Pipeline Parallelism for PyTorch
Python
648
star
35

kineto

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

torcharrow

High performance model preprocessing library on PyTorch
Python
635
star
37

tensordict

TensorDict is a pytorch dedicated tensor container.
Python
611
star
38

ios-demo-app

PyTorch iOS examples
Swift
588
star
39

tvm

TVM integration into PyTorch
C++
450
star
40

contrib

Implementations of ideas from recent papers
Python
390
star
41

ort

Accelerate PyTorch models with ONNX Runtime
Python
347
star
42

builder

Continuous builder and binary build scripts for pytorch
Shell
320
star
43

accimage

high performance image loading and augmenting routines mimicking PIL.Image interface
C
318
star
44

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.
Python
305
star
45

ao

Native PyTorch library for quantization and sparsity
Python
303
star
46

extension-ffi

Examples of C extensions for PyTorch
Python
256
star
47

nestedtensor

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.
Jupyter Notebook
252
star
48

tensorpipe

A tensor-aware point-to-point communication primitive for machine learning
C++
240
star
49

pytorch.github.io

The website for PyTorch
HTML
212
star
50

hydra-torch

Configuration classes enabling type-safe PyTorch configuration for Hydra apps
Python
198
star
51

workshops

This is a repository for all workshop related materials.
Jupyter Notebook
198
star
52

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.
Python
196
star
53

cppdocs

PyTorch C++ API Documentation
HTML
196
star
54

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.
C++
166
star
55

torchsnapshot

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

java-demo

Jupyter Notebook
126
star
57

torchdistx

Torch Distributed Experimental
Python
113
star
58

rfcs

PyTorch RFCs (experimental)
111
star
59

extension-script

Example repository for custom C++/CUDA operators for TorchScript
Python
111
star
60

csprng

Cryptographically secure pseudorandom number generators for PyTorch
Batchfile
100
star
61

pytorch_sphinx_theme

PyTorch Sphinx Theme
CSS
94
star
62

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
70
star
63

maskedtensor

MaskedTensors for PyTorch
Python
38
star
64

add-annotations-github-action

A GitHub action to run clang-tidy and annotate failures
JavaScript
13
star
65

ci-hud

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

probot

PyTorch GitHub bot written in probot
TypeScript
11
star
67

ossci-job-dsl

Jenkins job definitions for OSSCI
Groovy
9
star
68

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates
Makefile
5
star
69

torchhub_testing

Repo to test torchhub. Nothing to see here.
4
star
70

docs

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

dr-ci

Diagnose and remediate CI jobs
Haskell
2
star
72

pytorch-ci-dockerfiles

Scripts for generating docker images for PyTorch CI
2
star
73

labeler-github-action

GitHub action for labeling issues and pull requests based on conditions
TypeScript
1
star