bagua
Deep learning accelerator
A framework for accelerating PyTorch deep learning training
Bagua Speeds up PyTorch
877 stars
17 watching
83 forks
Language: Python
last commit: 4 months ago
Linked from 2 awesome lists
baguadeep-learningdistributeddistributed-computingdistributed-systemsmachine-learningpytorchrust-lang
ethicalml/awesome-production-machine-learning
zslucky/awesome-ai-booksRelated projects:
| Repository | Description | Stars |
|---|---|---|
 maxpumperla/elephas | Enables distributed deep learning with Keras and Spark for scalable model training | 1,574 |
 graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 569 |
 achaiah/pywick | A PyTorch-based neural network training framework with advanced features and utilities | 398 |
 furkanu/deeplearning.ai-pytorch | Re-implementation of Coursera's Deep Learning specialization assignments in PyTorch | 149 |
 open-mmlab/mmengine | Provides a flexible and configurable framework for training deep learning models with PyTorch. | 1,179 |
 zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,041 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,821 |
 eduardoleao052/js-pytorch | A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. | 1,084 |
 uber/petastorm | Enables training and evaluation of deep learning models from Apache Parquet datasets in various machine learning frameworks | 1,799 |
 vict0rsch/deep_learning | A collection of tutorials and resources on implementing deep learning models using Python libraries such as Keras and Lasagne. | 426 |
 dmmiller612/sparktorch | A PyTorch implementation on Apache Spark for distributed deep learning model training and inference. | 339 |
 rdspring1/pytorch_gbw_lm | Trains a large-scale PyTorch language model on the 1-Billion Word dataset | 123 |
 bamos/densenet.pytorch | An implementation of a deep neural network architecture in PyTorch | 832 |
 pyg-team/pytorch-frame | A deep learning framework for handling heterogeneous tabular data with diverse column types | 543 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |