A Python package for distributed machine learning training.
-
Training models made simple with Trainer and Train Configurations.
-
Support for gradient based and gradient free optimization.
-
Distributed training made simple across multi node/multi gpu setups.
-
Extensive callbacks, along with support for mlflow and tensorboard tracking.