pytorch
-
pytorch Public
Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
torchtitan Public
A PyTorch native platform for training generative AI models
-
executorch Public
On-device AI across mobile, embedded and edge for PyTorch
-
TensorRT Public
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
-
vision Public
Datasets, Transforms and Models specific to Computer Vision
-
audio Public
Data manipulation and transformation for audio signal processing, powered by PyTorch
-
test-infra Public
This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
-
rl Public
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.