12 projects
pytorch-ignite
A lightweight library to help with training neural networks in PyTorch.
torchvision
image and video datasets and models for torch deep learning
torch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
torchcodec
A video decoder for PyTorch
torchaudio
An audio package for PyTorch
torchtitan
A PyTorch native platform for training generative AI models
torchtune
A native-PyTorch library for LLM fine-tuning
torchtext
Text utilities, models, transforms, and datasets for PyTorch.
torchchat
Package for finetuning LLMs using native PyTorch
torchat
Package for finetuning LLMs using native PyTorch
torchcontrib
implementations of ideas from recent papers