clearml: A Python repository from reginepeltier

ClearML - Auto-Magical Suite of tools to streamline your ML workflow
Experiment Manager, ML-Ops and Data-Management

ClearML

Formerly known as Allegro Trains

ClearML is a ML/DL development and production suite, it contains three main modules:

Experiment Manager - Automagical experiment tracking, environments and results
ML-Ops - Automation, Pipelines & Orchestration solution for ML/DL jobs (K8s / Cloud / bare-metal)
Data-Management - Fully differentiable data management & version control solution on top of object-storage (S3/GS/Azure/NAS)

Instrumenting these components is the ClearML-server, see Self-Hosting & Free tier Hosting

Signup & Start using in under 2 minutes

ClearML Experiment Manager

Adding only 2 lines to your code gets you the following

Complete experiment setup log
- Full source control info including non-committed local changes
- Execution environment (including specific packages & versions)
- Hyper-parameters
  - ArgParser for command line parameters with currently used values
  - Explicit parameters dictionary
  - Tensorflow Defines (absl-py)
  - Hydra configuration and overrides
- Initial model weights file
Full experiment output automatic capture
- stdout and stderr
- Resource Monitoring (CPU/GPU utilization, temperature, IO, network, etc.)
- Model snapshots (With optional automatic upload to central storage: Shared folder, S3, GS, Azure, Http)
- Artifacts log & store (Shared folder, S3, GS, Azure, Http)
- Tensorboard/TensorboardX scalars, metrics, histograms, images, audio and video samples
- Matplotlib & Seaborn
- ClearML Explicit Logging interface for complete flexibility.
Extensive platform support and integrations
- Supported ML/DL frameworks: PyTorch(incl' ignite/lightning), Tensorflow, Keras, AutoKeras, XGBoost and Scikit-Learn
- Seamless integration (including version control) with Jupyter Notebook and PyCharm remote debugging

Start using ClearML

pip install clearml

Add two lines to your code:

from clearml import Task
task = Task.init(project_name='examples', task_name='hello world')

You are done, everything your process outputs is now automagically logged into ClearML.
Next step automation! Learn more on ClearML two clicks automation here

ClearML Architecture

The ClearML run-time components:

The ClearML Python Package for integrating ClearML into your existing scripts by adding just two lines of code, and optionally extending your experiments and other workflows with ClearML powerful and versatile set of classes and methods.
The ClearML Server storing experiment, model, and workflow data, and supporting the Web UI experiment manager, and ML-Ops automation for reproducibility and tuning. It is available as a hosted service and open source for you to deploy your own ClearML Server.
The ClearML Agent for ML-Ops orchestration, experiment and workflow reproducibility, and scalability.

Additional Modules

clearml-session - Launch remote JupyterLab / VSCode-server inside any docker, on Cloud/On-Prem machines
clearml-task - Run any codebase on remote machines with full remote logging of Tensorboard, Matplotlib & Console outputs
clearml-data - CLI for managing and versioning your datasets, including creating / uploading / downloading of data from S3/GS/Azure/NAS
AWS Auto-Scaler - Automatically spin EC2 instances based on your workloads with preconfigured budget! No need for K8s!
Hyper-Parameter Optimization - Optimize any code with black-box approach and state of the art Bayesian optimization algorithms
Automation Pipeline - Build pipelines based on existing experiments / jobs, supports building pipelines of pipelines!
Slack Integration - Report experiments progress / failure directly to Slack (fully customizable!)

Why ClearML?

ClearML is our solution to a problem we share with countless other researchers and developers in the machine learning/deep learning universe: Training production-grade deep learning models is a glorious but messy process. ClearML tracks and controls the process by associating code version control, research projects, performance metrics, and model provenance.

We designed ClearML specifically to require effortless integration so that teams can preserve their existing methods and practices.

Use it on a daily basis to boost collaboration and visibility in your team
Create a remote job from any experiment with a click of a button
Automate processes and create pipelines to collect your experimentation logs, outputs, and data
Store all you data on any object-storage solution, with the simplest interface possible
Make you data transparent by cataloging it all on the ClearML platform

We believe ClearML is ground-breaking. We wish to establish new standards of true seamless integration between experiment management,ML-Ops and data management.

Who We Are

ClearML is supported by the team behind allegro.ai, where we build deep learning pipelines and infrastructure for enterprise companies.

We built ClearML to track and control the glorious but messy process of training production-grade deep learning models. We are committed to vigorously supporting and expanding the capabilities of ClearML.

We promise to always be backwardly compatible, making sure all your logs, data and pipelines will always upgrade with you.

License

Apache License, Version 2.0 (see the LICENSE for more information)

Documentation, Community & Support

More information in the official documentation and on YouTube.

For examples and use cases, check the examples folder and corresponding documentation.

If you have any questions: post on our Slack Channel, or tag your questions on stackoverflow with 'clearml' tag (previously trains tag).

For feature requests or bug reports, please use GitHub issues.

Additionally, you can always find us at clearml@allegro.ai

Contributing

See the ClearML Guidelines for Contributing.

May the force (and the goddess of learning rates) be with you!

reginepeltier/clearml