xla

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

created at Nov. 5, 2018, 10:42 p.m.

57

2,460

469

executorch

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

created at Feb. 25, 2022, 5:58 p.m.

61

2,312

401

torchtune

A Native-PyTorch Library for LLM Fine-tuning

created at Oct. 20, 2023, 9:10 p.m.

45

3,902

350

torchtitan

A native PyTorch Library for large model training

created at Dec. 13, 2023, 1:51 a.m.

44

2,752

220

cpuinfo

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

created at March 3, 2017, 7:46 a.m.

247

951

298

torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

created at March 22, 2024, 6:15 p.m.

34

2,571

160

rl

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

created at Feb. 1, 2022, 5:17 p.m.

41

2,114

281

ao

PyTorch native quantization and sparsity for training and inference

created at Nov. 3, 2023, 9:27 p.m.

41

1,578

173

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

created at Aug. 13, 2016, 5:26 a.m.

1,744

84,405

22,734

examples

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

created at Aug. 24, 2016, 3:12 a.m.

401

21,475

9,398

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

created at Aug. 29, 2017, 4:26 p.m.

339

16,687

4,398

ignite

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

created at Nov. 23, 2017, 5:31 p.m.

62

3,531

457

contrib

Implementations of ideas from recent papers

created at July 11, 2018, 2:38 p.m.

21

108

0

ELF

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

created at April 26, 2018, 9:33 p.m.

171

2,650

442

glow

Compiler for Neural Network hardware accelerators

created at Sept. 29, 2017, 2:28 p.m.

146

2,425

527

QNNPACK

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

created at Oct. 22, 2018, 8:01 p.m.

88

1,143

150

FBGEMM

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

created at Sept. 24, 2018, 7:07 p.m.

55

1,135

457

botorch

Bayesian optimization in PyTorch

created at July 30, 2018, 11:59 p.m.

48

1,281

88

tvm

TVM integration into PyTorch

created at March 28, 2019, 10:39 p.m.

14

124

10

hub

None

created at April 22, 2019, 4:51 p.m.

42

676

49

audio

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

created at May 5, 2017, 12:38 a.m.

68

1,796

462

vision

Datasets, Transforms and Models specific to Computer Vision

created at Nov. 9, 2016, 11:11 p.m.

432

16,082

6,939

captum

Model interpretability and understanding for PyTorch

created at Aug. 27, 2019, 3:34 p.m.

183

2,665

282

elastic

PyTorch elastic training

created at Oct. 1, 2019, 9:48 p.m.

22

213

8

serve

Serve, optimize and scale PyTorch models in production

created at Oct. 3, 2019, 3:17 a.m.

57

4,209

859

text

Data loaders and abstractions for text and NLP

created at Dec. 12, 2016, 12:56 a.m.

260

3,106

752

opacus

Training PyTorch models with differential privacy

created at Dec. 7, 2019, 1:58 a.m.

27

427

55

tutorials

PyTorch tutorials.

created at Sept. 30, 2016, 11:48 p.m.

171

6,637

3,590

android-demo-app

PyTorch android examples of usage in applications

created at Sept. 27, 2019, 3:11 a.m.

22

611

265

builder

Continuous builder and binary build scripts for pytorch

created at Sept. 2, 2016, 5:02 p.m.

14

163

117

data

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

created at May 12, 2021, 8:55 p.m.

16

572

31

live

PyTorch Live is an easy to use library of tools for creating on-device ML demos on Android and iOS. With Live, you can build a working mobile app ML demo in minutes.

created at Nov. 12, 2021, 6 p.m.

20

132

11

torchrec

Pytorch domain library for recommendation systems

created at July 12, 2021, 11:15 p.m.

24

1,240

201

functorch

functorch is JAX-like composable function transforms for PyTorch.

created at April 19, 2021, 11:24 p.m.

18

759

48

torchdynamo

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

created at Sept. 3, 2021, 3:31 a.m.

37

583

183