pytorch
-
pytorch Public
Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
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.
-
executorch Public
On-device AI across mobile, embedded and edge for PyTorch
-
helion Public
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
-
torchtitan Public
A PyTorch native platform for training generative AI models
-
ignite Public
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
-
ao Public
PyTorch native quantization and sparsity for training and inference
-
TensorRT Public
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT