Skip to main content

An AutoML Library made with Optuna and PyTorch Lightning

Project description

logo

An AutoML Library made with Optuna and PyTorch Lightning

Maintainability Rating Reliability Rating Security Rating Coverage Documentation Status DOI Slack

PyPI - Python Version PyPI version Downloads Downloads license

Installation

Recommended

pip install -U gradsflow

From source

pip install git+https://github.com/gradsflow/gradsflow@main

Highlights

What is Gradsflow?

Gradsflow is based on Optuna and PyTorch Lightning ⚡️. It leverages PyTorch Lightning Flash so that you don't have to write any PyTorch or Optuna code for model building or hyperparameter tuning 🚀

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.

📑 Check out notebooks examples.

💬 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

Acknowledgement

Gradsflow is built with help of Optuna and PyTorch Lightning 💜

Download files

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

Source Distribution

gradsflow-0.0.2.tar.gz (26.5 kB view hashes)

Uploaded Source

Built Distribution

gradsflow-0.0.2-py3-none-any.whl (16.7 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page