PyPI page
Home page
Author:
Kuaishou AI Platform & DS3 Lab
License:
Summary:
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.
Latest version:
0.9.2
Required dependencies:
colorama
|
deprecation
|
flask
|
gevent
|
gorilla
|
numpy
|
parallel-ssh
|
prometheus-client
|
pydantic
|
pytest-benchmark
|
requests
|
scikit-learn
|
scikit-optimize
|
setuptools-rust
|
tqdm
|
xxhash
Downloads last day:
0
Downloads last week:
18
Downloads last month:
76