Aim

3.29.1 · active · verified Thu Apr 16

Aim is an open-source, self-hosted AI metadata and ML experiment tracking tool that helps record, search, and compare AI experiments. It provides a performant UI for exploring and comparing runs and an SDK for programmatic access to tracked metadata. As of version 3.29.1, it continues to release frequent minor updates and bug fixes, typically multiple times per year, within its major version. [1, 12, 18]

Common errors

Warnings

Install

Imports

Quickstart

This quickstart demonstrates how to initialize an Aim run, log hyperparameters, and track metrics. After running this script, navigate to your project directory in the terminal and run `aim up` to launch the Aim UI and visualize your experiment results. [1, 12]

import aim
import math

# Initialize a new run
run = aim.Run()

# Log hyper-parameters
run["hparams"] = {
    "learning_rate": 0.001,
    "batch_size": 32,
    "optimizer": "Adam"
}

# Log metrics
for step in range(100):
    run.track(math.sin(step / 10.0), name='sine', step=step)
    run.track(math.cos(step / 10.0), name='cosine', step=step)

print(f"Aim run created with hash: {run.hash}. Start UI with 'aim up'.")

view raw JSON →