A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
BSD-3-CLAUSE License
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
Install PyTorch distributions with computation backend auto-detection
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iter...
TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
A powerful and flexible machine learning platform for drug discovery
High performance model preprocessing library on PyTorch
Datasets, Transforms and Models specific to Computer Vision
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Pytorch domain library for recommendation systems
Rust bindings for the C++ api of PyTorch.
TensorDict is a pytorch dedicated tensor container.
A performant, memory-efficient checkpointing library for PyTorch applications, designed with larg...
torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem b...
Torch Distributed Experimental