bagua. 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.

github.com/Youhe-Jiang/bagua

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.