• 3
• 0
• 0

helion

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

• 2
• 0
• 0

torchft

Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)

• 3
• 0
• 0

rl

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

• 2
• 0
• 0

ci-infra

• 2
• 0
• 0

expecttest

• 2
• 0
• 1

torchcodec

PyTorch media decoding and encoding

• 2
• 0
• 0

torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

• 3
• 0
• 0

tau

Pipeline Parallelism for PyTorch

• 3
• 0
• 0

torcharrow

High performance model preprocessing library on PyTorch

• 2
• 0
• 0

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

• 2
• 0
• 0

csprng

Cryptographically secure pseudorandom number generators for PyTorch

• 2
• 0
• 0

opacus

Training PyTorch models with differential privacy

• 2
• 0
• 0

tensorpipe

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

• 2
• 0
• 0

nestedtensor

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

• 2
• 0
• 0

elastic

PyTorch elastic training

• 2
• 0
• 0

QNNPACK

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

• 2
• 0
• 0

ossci-job-dsl

Jenkins job definitions for OSSCI

• 2
• 0
• 0

examples

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

• 2
• 0
• 0

torchx

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E...

• 3
• 0
• 0

PiPPy

Pipeline Parallelism for PyTorch