4

GitHub - gradsflow/gradsflow: An open-source AutoML Library in PyTorch

 2 years ago
source link: https://github.com/gradsflow/gradsflow
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

An open-source AutoML Library in PyTorch

Installation

Recommended

pip install -U gradsflow

From source

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

Highlights

What is GradsFlow?

!!! attention GradsFlow is changing fast and is not stable yet.

GradsFlow is based on Ray and PyTorch Lightning zap (support for other torch frameworks will be added soon). It leverages PyTorch Lightning Flash so that you don't have to write any PyTorch or Optuna code for model building or hyperparameter tuning rocket

Although you might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch/Lightning Flash Model with Gradsflow AutoModel sparkles

  • 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.

bookmark_tabs Check out notebooks examples.

speech_balloon Join the Slack group to chat with us.

hugs 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 Ray and PyTorch Lightning purple_heart


Recommend

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK