Stars
A lightweight library for PyTorch training tools and utilities
TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.
A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools…
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.
A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
Pytorch domain library for recommendation systems
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Hackable and optimized Transformers building blocks, supporting a composable construction.
Models, data loaders and abstractions for language processing, powered by PyTorch
Datasets, Transforms and Models specific to Computer Vision
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.