• 2
• 0
• 0

helion

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

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

• 2
• 0
• 0

tlparse

TORCH_LOGS parser for PT2

• 2
• 0
• 0

gloo

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

• 2
• 0
• 1

torchcodec

PyTorch media decoding and encoding

• 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

torchchat

Run PyTorch LLMs locally on servers, desktop and mobile

• 2
• 0
• 0

expecttest

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

• 3
• 0
• 0

tau

Pipeline Parallelism for PyTorch

• 3
• 0
• 0

torcharrow

High performance model preprocessing library on PyTorch

• 3
• 0
• 0

maskedtensor

MaskedTensors for PyTorch

• 3
• 0
• 0

live

PlayTorch is a framework for rapidly creating mobile AI experiences.

• 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

kineto

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

• 2
• 0
• 0

hydra-torch

Configuration classes enabling type-safe PyTorch configuration for Hydra apps

• 2
• 0
• 0

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

• 2
• 0
• 0

opacus

Training PyTorch models with differential privacy