xla

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

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

56

2,121

368

executorch

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

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

48

833

158

torchtune

A Native-PyTorch Library for LLM Fine-tuning

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

36

2,438

158

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

1,702

77,846

21,038

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.

31

434

121

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.

408

15,376

6,839

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

3,879

788

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