Skip to content


An open-source AutoML Library in PyTorch

Maintainability Rating Reliability Rating Security Rating codecov Documentation Status DOI Slack

PyPI - Python Version PyPI version Downloads Downloads license


pip install -U gradsflow

From source

pip install git+


What is GradsFlow?


GradsFlow is changing fast and is not stable yet.

GradsFlow is an open-source AutoML Library for PyTorch that can train Deep Learning Models on your laptop or to a remote cluster directly from your laptop. Our aim is to enable non ML expert to train and build AI Products. GradsFlow leverages the power of PyTorch Lightning ⚑️ and Ray️. It leverages PyTorch Lightning Flash so that you don't have to write any code for model building or hyperparameter tuning πŸš€

GradsFlow Model API provides a simple Keras like model training functionality πŸ”₯. You can train any PyTorch model using and it is easily customizable for more complex tasks.

You might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch Model with Gradsflow AutoModel ✨

  • gradsflow.core: Core defines the building blocks of AutoML tasks.

  • gradsflow.autotasks: AutoTasks defines different ML/DL tasks which is provided by gradsflow AutoML API.

  • gradsflow.model: Model defines the model training functionality.

πŸ“‘ Check out notebooks examples.

πŸ’¬ Join the Slack group to chat with us.

πŸ’™ Sponsor us on ko-fi

πŸ“§ Do you need support? Contact us at

πŸ€— 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


GradsFlow is built with help of awesome open-source projects (including but not limited to) PyTorch Lightning and Ray πŸ’œ

It takes inspiration from multiple APIs like Keras, FastAI.

Last update: September 26, 2021