Kaen 🔥火炎🔥
is a toolkit for delightful deep learning with PyTorch in public clouds.
Kaen simplifies
- setup of PyTorch distributed training environments in AWS, Azure, and GCP
- containerization of PyTorch models using Docker
- integration of Optuna hyperparameter tuning with PyTorch
- experiment tracking with MLFlow
Fun Facts