Skip to main content

An open-source AutoML Library based on PyTorch

Project description

An open-source AutoML & PyTorch Model Training Library

Docs | Examples

pytest Documentation Status codecov PyPI version Downloads Slack



Recommended: pip install -U gradsflow

Latest (unstable): pip install git+

About GradsFlow

!!! attention 🚨 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 provides a powerful and easy-to-extend Model Training API that can be used to train almost any PyTorch model. Though GradsFlow has its own Model Training API it also supports PyTorch Lightning Flash to provide more rich features across different tasks. It takes inspiration & leverages some cool OSS library including Ray️, HuggingFace Accelerate, PyTorch Lightning, Keras &

  • 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. You can update documentation, add examples, fix identified issues, add/request a new feature.

For more details check out 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) Ray, PyTorch Lightning, HuggingFace Accelerate, TorchMetrics. It takes inspiration from multiple projects Keras & FastAI.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for gradsflow, version 0.0.6.post1
Filename, size File type Python version Upload date Hashes
Filename, size gradsflow-0.0.6.post1-py3-none-any.whl (49.7 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size gradsflow-0.0.6.post1.tar.gz (1.1 MB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page