• 2
• 0
• 0

test-infra

This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track d...

• 3
• 0
• 0

helion

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

• 2
• 0
• 0

gloo

Collective communications library with various primitives for multi-machine training.

• 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
• 1

torchcodec

PyTorch media decoding and encoding

• 2
• 0
• 0

expecttest

• 2
• 0
• 0

torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

• 3
• 0
• 0

tau

Pipeline Parallelism for PyTorch

• 3
• 0
• 0

torcheval

A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate me...

• 2
• 0
• 0

kineto

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

• 3
• 1
• 0

torcharrow

High performance model preprocessing library on PyTorch

• 3
• 0
• 0

maskedtensor

MaskedTensors for PyTorch

• 3
• 0
• 0

ci-hud

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

• 2
• 0
• 0

opacus

Training PyTorch models with differential privacy

• 2
• 0
• 0

csprng

Cryptographically secure pseudorandom number generators for PyTorch

• 2
• 0
• 0

hydra-torch

Configuration classes enabling type-safe PyTorch configuration for Hydra apps

• 2
• 0
• 0

dr-ci

Diagnose and remediate CI jobs

• 2
• 0
• 0

tensorpipe

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