PyTorch extensions for high performance and large scale training.
OTHER License
The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video ...
Collection of common code that's shared among different research projects in FAIR computer vision...
An open-source AutoML Library based on PyTorch
High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...
Deep Learning and Reinforcement Learning Library for Scientists and Engineers
A collection of computer vision pre-trained models.
š A simple way to launch, train, and use PyTorch models on almost any device and distributed conf...
š¤ PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
Unsupervised Language Modeling at scale for robust sentiment classification
Hackable and optimized Transformers building blocks, supporting a composable construction.
An Easy-to-use, Scalable and High-performance RLHF Framework (70B+ PPO Full Tuning & Iterative DP...
The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relat...
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
In this repository, I will share some useful notes and references about deploying deep learning-b...