Bagua Python Packages

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.

persia

High performance distributed framework for training deep learning recommendation models based on PyTorch.

bagua-cuda111

Bagua Speeds up PyTorch

bagua-cuda113

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.

bagua-cuda115

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.

bagua-cuda102

Bagua Speeds up PyTorch

bagua-cuda116

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.

persia-cuda102

PersiaML Python Library

persia-cuda111

High performance distributed framework for training deep learning recommendation models based on PyTorch.

persia-cuda113

High performance distributed framework for training deep learning recommendation models based on PyTorch.

bagua-cuda117

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.