An AutoML Library made with Optuna and PyTorch Lightning
Installation
Recommended
pip install -U gradsflow
From source
pip install git+https://github.com/gradsflow/gradsflow@main
Highlights
- 2020-8-25: Released first version 0.0.1
โจ ๐
What is Gradsflow?
Gradsflow is based on Optuna and PyTorch Lightning
Although you might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch/Lightning Flash Model with Gradsflow AutoModel
-
gradsflow.core
: Core defines the building blocks of AutoML tasks. -
gradsflow.taskauto
: AutoTasks defines different ML/DL tasks which is provided by Gradsflow AutoML API.
๐ค
Contribute
Contributions of any kind are welcome. Please check the Contributing Guidelines before contributing.
Code Of Conduct
We pledge to act and interact in ways that contribute to an open, welcoming, diverse, inclusive, and healthy community.
Read full Contributor Covenant Code of Conduct
Acknowledgement
Gradsflow is built with help of Optuna and PyTorch Lightning