Skip to main content

An open-source AutoML Library in PyTorch

Project description

logo

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

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 an open-source AutoML Library for PyTorch that can train Deep Learning Models on your laptop or to a remote cluster directly from your laptop. Our aim is to enable non ML expert to train and build AI Products. GradsFlow leverages the power of PyTorch Lightning ⚡️ and Ray️. It leverages PyTorch Lightning Flash so that you don't have to write any code for model building or hyperparameter tuning 🚀

GradsFlow Model API provides a simple Keras like model training functionality 🔥. You can train any PyTorch model using model.fit(...) and it is easily customizable for more complex tasks.

You might want to train a custom model and search hyperparameters, You can easily integrate any PyTorch Model with Gradsflow AutoModel

  • 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: Model defines the model training functionality.

📑 Check out notebooks examples.

💬 Join the Slack group to chat with us.

💙 Sponsor us on ko-fi

📧 Do you need support? Contact us at admin@gradsflow.com

🤗 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 awesome open-source projects (including but not limited to) PyTorch Lightning and Ray 💜

It takes inspiration from multiple APIs like Keras, FastAI.

Project details


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.5.post0.tar.gz (87.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

gradsflow-0.0.5.post0-py3-none-any.whl (34.1 kB view details)

Uploaded Python 3

File details

Details for the file gradsflow-0.0.5.post0.tar.gz.

File metadata

  • Download URL: gradsflow-0.0.5.post0.tar.gz
  • Upload date:
  • Size: 87.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.2 importlib_metadata/4.8.1 pkginfo/1.7.1 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.7

File hashes

Hashes for gradsflow-0.0.5.post0.tar.gz
Algorithm Hash digest
SHA256 c64be1de8f05b7e3c57e90de81d06e5c4b1136fd74e7b7989d081118755632d9
MD5 0d74cf7d719d3afdad70ca8cc5bf982b
BLAKE2b-256 2757c1fc965d082d173022aba9e2f5a59ba59566c9086d43c43dedf4820f1b66

See more details on using hashes here.

File details

Details for the file gradsflow-0.0.5.post0-py3-none-any.whl.

File metadata

  • Download URL: gradsflow-0.0.5.post0-py3-none-any.whl
  • Upload date:
  • Size: 34.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.2 importlib_metadata/4.8.1 pkginfo/1.7.1 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.7

File hashes

Hashes for gradsflow-0.0.5.post0-py3-none-any.whl
Algorithm Hash digest
SHA256 47dc176092af219415ec5ecdc02d70582713388ebdbd7a3cfea3026e5885822d
MD5 a94844d298e5162e63f5aed654936f4a
BLAKE2b-256 797a36b0329feb9907500fa0a1f68ca10b0dbaa671b58adf939d0fba527116a0

See more details on using hashes here.

Supported by

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