Aim package logs your training runs, enables a beautiful UI to compare them and an API to query them programmatically.
Why use Aim?
Modern ML development revolves around collection and analysis of AI metadata (training metrics, images, distributions etc) to analyze and explore different aspects of the model performance.
There is both a need to manually explore and compare the metadata as well as automate for different infrastructure needs.
Aim helps to track AI metadata and
Explore it manually through the most advanced open-source experiment comparison web UI.
Query programmatically in your favorite notebook or through script for automation.
Use Aim to seamlessly log your ML metadata in your training environment and explore through UI and code. Aim is free, open-source and self-hosted.
What can you do with Aim?
Log metrics and params
Use the Aim SDK to log as many metrics and params as you need for your training and evaluation runs. Aim users track 1000s of training runs and sometimes more than 100s of metrics per run with lots of steps.
Query metadata on Web UI
Aim enables a powerful pythonic query language to filter through metadata. It’s like a python if statement over everything you have tracked. You can use this on all explorer screens.
Runs explorer will help you to hollistically view all your runs, each metric last tracked values and tracked hyperparameters.
Metrics explorer helps you to compare 100s of metrics within a few clicks. It helps to save lots of time compared to other open-source experiment tracking tools.
Params explorer enables a parallel coordinates view for metrics and params. Very helpful when doing hyperparameter search.
Images explorer (coming soon…)
Track intermediate images and search, compare them on the Images Explorer.
Query metadata programmatically
Use the same pythonic if statement to query the data through the Aim SDK programmatically.
How Aim works?
Aim is a python package with three main components:
A rocksdb-based embedded storage where the metadata is stored locally
A simple python interface that allows to track AI metadata
images (coming soon…)
distributions (coming soon…)
A self-hosted web interface to deeply explore the tracked metadata
Integrated with your favorite tools
If you have questions please: