Bagua is a deep learning training acceleration framework for PyTorch. It provides a one-stop training acceleration solution, including faster distributed training compared to PyTorch DDP, faster dataloader, kernel fusion, and more. [wheel] | |
https://github.com/BaguaSys/bagua | |
bagua-cuda115-0.9.0 | |
download~amd64 ~x86 | pypi |
bagua-cuda115-0.8.3_pre213 | |
download~amd64 ~x86 | pypi |