fairscale
Training framework
A PyTorch extension library that provides high-performance and large-scale training techniques.
PyTorch extensions for high performance and large scale training.
3k stars
50 watching
282 forks
Language: Python
last commit: 2 months ago
Linked from 1 awesome list
amrzv/awesome-colab-notebooksRelated projects:
| Repository | Description | Stars |
|---|---|---|
 nvidia/apex | Tools for streamlined mixed precision and distributed training in PyTorch | 8,460 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 84,978 |
| facebookresearch/pytorch3d | A deep learning library for 3D data processing and computer vision research using PyTorch | 8,889 |
| pytorch/ignite | A high-level library to help with training and evaluating neural networks in PyTorch | 4,554 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,530 |
 bytedance/byteps | A high-performance distributed deep learning framework supporting multiple frameworks and networks | 3,635 |
 victoresque/pytorch-template | An open-source PyTorch deep learning project template providing a structured environment for building and training various models. | 4,784 |
 huggingface/accelerate | A tool to simplify training and deployment of PyTorch models on various devices and configurations | 8,056 |
| facebookresearch/fairseq | A toolkit for training custom sequence-to-sequence models for various NLP tasks | 30,675 |
 qubvel-org/segmentation_models.pytorch | A comprehensive library for training and applying deep learning models for image segmentation | 9,829 |
 mlfoundations/open_flamingo | A framework for training large multimodal models to generate text conditioned on images or other text. | 3,781 |
| facebookresearch/slowfast | Provides state-of-the-art video understanding codebase with efficient training methods and pre-trained models for various tasks | 6,680 |
| facebookresearch/dinov2 | A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision. | 9,425 |
 jfzhang95/pytorch-deeplab-xception | A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets | 2,919 |
 sktime/pytorch-forecasting | A PyTorch-based package for state-of-the-art time series forecasting with deep learning architectures | 4,038 |