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. There will be a lot of breaking changes until we reach 0.1.0. Feel free to give your feedback by creating an issue or join our Slack group.

GradsFlow is an open-source AutoML Library based on PyTorch. The goal of GradsFlow is to democratize AI and make it available to everyone.

It can automatically train Deep Learning Models for different tasks on your laptop or to a remote cluster directly from your laptop. It also provides a powerful Model Training API that can be used to train almost any PyTorch model. GradsFlow leverages some cool OSS library including Ray️ and PyTorch Lightning. You don't have to write any PyTorch or hyperparameter optimization code.

  • 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: GradsFlow Model provides a simple and yet customizable Model Training API. You can train any PyTorch model using and it is easily customizable for more complex tasks.

  • gradsflow.tuner: AutoModel HyperParameter search with minimal code changes.

πŸ“‘ Check out notebooks examples to learn more.

πŸ’™ Sponsor on ko-fi

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


Stay Up-to-Date

Social: You can also follow us on Twitter @gradsflow and [Linkedin][ for the latest updates.

Questions & Discussion

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

πŸ€— 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: October 3, 2021