https://github.com/gradsflow/gradsflow-automl
An open-source AutoML Library based on PyTorch
https://github.com/gradsflow/gradsflow-automl
artificial-intelligence autoai automl deeplearning machine-learning python pytorch
Last synced: 6 months ago
JSON representation
An open-source AutoML Library based on PyTorch
- Host: GitHub
- URL: https://github.com/gradsflow/gradsflow-automl
- Owner: gradsflow
- License: apache-2.0
- Created: 2021-08-11T17:37:54.000Z (almost 5 years ago)
- Default Branch: main
- Last Pushed: 2025-04-07T17:45:47.000Z (about 1 year ago)
- Last Synced: 2025-04-13T00:55:28.174Z (about 1 year ago)
- Topics: artificial-intelligence, autoai, automl, deeplearning, machine-learning, python, pytorch
- Language: Python
- Homepage: https://docs.gradsflow.com/
- Size: 3.87 MB
- Stars: 307
- Watchers: 10
- Forks: 36
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- Funding: .github/FUNDING.yml
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
- Citation: CITATION.cff
- Codeowners: .github/CODEOWNERS
Awesome Lists containing this project
README
An open-source AutoML & PyTorch Model Training Library
---
[][#github-license]
[][#pytest-package]
[](https://gradsflow.readthedocs.io/en/latest/?badge=latest)
[](https://codecov.io/gh/gradsflow/gradsflow)
[](https://badge.fury.io/py/gradsflow)
[][#pypi-package]
[](https://pepy.tech/project/gradsflow)
[][#conda-forge-package]
[][#conda-forge-package]
[][#slack-package]
[#github-license]: https://github.com/gradsflow/gradsflow/blob/main/LICENSE
[#pypi-package]: https://pypi.org/project/gradsflow/
[#conda-forge-package]: https://anaconda.org/conda-forge/gradsflow
[#slack-package]: https://join.slack.com/t/gradsflow/shared_invite/zt-ulc0m0ef-xstzyowuTgYceVmFbJlBmg
[#pytest-package]: https://github.com/gradsflow/gradsflow/actions/workflows/main.yml
## Highlights
- 2021-10-7: [v0.0.6 Release blog post](https://towardsdatascience.com/gradsflow-democratizing-ai-with-automl-9a8a75d6b7ea)
- 2021-10-5: [Hacktoberfest 2021 Kickoff event](https://youtu.be/lVtxXyCAZ-4?t=2647)
- 2021-10-4: Model Trainer support
- 2021-8-29: Migrated to Ray Tune
- 2021-8-25: [Released first version 0.0.1](https://pypi.org/project/gradsflow/) ✨ :tada:
## 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](https://join.slack.com/t/gradsflow/shared_invite/zt-ulc0m0ef-xstzyowuTgYceVmFbJlBmg) group.
GradsFlow is an open-source AutoML Library based on PyTorch.
Our goal is to democratize AI and make it available to everyone.
It can automatically build & 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](https://docs.gradsflow.com/en/latest/gradsflow/models/model/)
that can be used to train almost any PyTorch model.
Though GradsFlow has its own Model Training API it also supports [PyTorch Lightning Flash](https://lightning-flash.readthedocs.io/en/latest)
to provide more rich features across different tasks.
!!! info
Gradsflow is built for both *beginners* and *experts*! `AutoTasks` provides zero-code AutoML while
`Model` and `Tuner` provides custom model training and Hyperparameter optimization.
### Installation
**Recommended**:
The recommended method of installing `gradsflow` is either with `pip` from PyPI or, with `conda` from conda-forge channel.
- **with pip**
```sh
pip install -U gradsflow
```
- **with conda**
```sh
conda install -c conda-forge gradsflow
```
**Latest** (unstable):
You can also install the latest bleeding edge version (could be unstable) of `gradsflow`, should you feel motivated enough, as follows:
```sh
pip install git+https://github.com/gradsflow/gradsflow@main
```
### Automatic Model Building and Training
Are you a beginner or from non Machine Learning background? This section is for you. Gradsflow `AutoTask` provides
automatic model building and training across various different tasks
including Image Recognition, Sentiment Analysis, Text Summarization and more to come.
### Simplified Hyperparameter tuning API
`Tuner` provides a simplified API to move from Model Training to Hyperparameter optimization.
### Components
- `gradsflow.core`: [Core](https://docs.gradsflow.com/en/latest/gradsflow/core/) defines the building blocks
of AutoML tasks.
- `gradsflow.autotasks`: [AutoTasks](https://docs.gradsflow.com/en/latest/gradsflow/tasks/) defines
different ML/DL tasks which is provided by gradsflow AutoML API.
- `gradsflow.model`: GradsFlow [Model](https://docs.gradsflow.com/en/latest/gradsflow/models/model/) provides a simple and
yet customizable Model Training API.
You can train any PyTorch model using `model.fit(...)` and it is easily customizable for more complex tasks.
- `gradsflow.tuner`: [AutoModel](https://docs.gradsflow.com/en/latest/gradsflow/tuner/) HyperParameter search with minimal code changes.
📑 Check out [notebooks examples](https://github.com/gradsflow/gradsflow/tree/main/examples/nbs) to learn more.
🧡 Sponsor on [ko-fi](https://ko-fi.com/aniketmaurya)
📧 Do you need support? Contact us at
## Community
### Stay Up-to-Date
**Social**: You can also follow us on Twitter [@gradsflow](https://twitter.com/gradsflow) and [Linkedin](https://www.linkedin.com/company/gradsflow) for the latest updates.
### Questions & Discussion
💬 Join the [Slack](https://join.slack.com/t/gradsflow/shared_invite/zt-ulc0m0ef-xstzyowuTgYceVmFbJlBmg) 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**](https://github.com/gradsflow/gradsflow/blob/master/CONTRIBUTING.md) 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**](https://github.com/gradsflow/gradsflow/blob/master/CODE_OF_CONDUCT.md)
## Acknowledgement
**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](https://keras.io) & [FastAI](https://docs.fast.ai).