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-cuda116-0.9.3_pre6 | |
download~amd64 ~x86 | pypi |
bagua-cuda116-0.9.2 | |
download~amd64 ~x86 | pypi |