9 projects
torchvision
image and video datasets and models for torch deep learning
torch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
triton
A language and compiler for custom Deep Learning operations
executorch
On-device AI across mobile, embedded and edge for PyTorch
torchao
Package for applying ao techniques to GPU models
torchaudio
An audio package for PyTorch
torchtune
A native-PyTorch library for LLM fine-tuning
torchdata
Composable data loading modules for PyTorch
torchtext
Text utilities, models, transforms, and datasets for PyTorch.
