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
48 watching
281 forks
Language: Python
last commit: 8 days 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,407 |
 pytorch/pytorch | A Python library providing tensors and dynamic neural networks with strong GPU acceleration | 83,959 |
| facebookresearch/pytorch3d | A deep learning library for 3D data processing and computer vision research using PyTorch | 8,806 |
| pytorch/ignite | A high-level library to help with training and evaluating neural networks in PyTorch | 4,526 |
| pytorch/examples | A collection of curated examples showcasing various PyTorch applications in computer vision, natural language processing, and reinforcement learning. | 22,428 |
 bytedance/byteps | A high-performance distributed deep learning framework supporting multiple frameworks and networks | 3,630 |
 victoresque/pytorch-template | An open-source PyTorch deep learning project template providing a structured environment for building and training various models. | 4,754 |
 huggingface/accelerate | A tool to simplify training and deployment of PyTorch models on various devices and configurations | 7,947 |
| facebookresearch/fairseq | A toolkit for training custom sequence-to-sequence models for various NLP tasks | 30,522 |
 qubvel-org/segmentation_models.pytorch | A PyTorch library for building and training neural networks for image segmentation tasks. | 9,696 |
 mlfoundations/open_flamingo | A framework for training large multimodal models to generate text conditioned on images or other text. | 3,742 |
| facebookresearch/slowfast | Provides state-of-the-art video understanding codebase with efficient training methods and pre-trained models for various tasks | 6,623 |
| facebookresearch/dinov2 | A PyTorch implementation of a self-supervised learning method for learning robust visual features without supervision. | 9,211 |
 jfzhang95/pytorch-deeplab-xception | A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets | 2,910 |
 sktime/pytorch-forecasting | A PyTorch-based package for state-of-the-art time series forecasting with deep learning architectures | 4,001 |