An easy-to-use & supercharged open-source AI metadata tracker

Aim logs all your AI metadata (experiments, prompts, etc) enables a UI to compare & observe them and SDK to query them programmatically.

Check out our Github

Trusted by ML teams from

meta
amazon
microsoft
ibm
airbus
amd
mila
mit

Seamless integrations with your favorite tools

Aim is tightly integrated with the ML ecosystem. Aim comes with built-in callbacks for the most of ML tools and frameworks. It allows to track metadata (metrics, hparams, etc.) with just a few lines of code!

Learn more
integrations image

Get started
in under a minute and on any environment

pip install aim
import aim
import math

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

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

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

Now check out our GitHub repo or documentation to learn more

Why use Aim?

Aim is an open-source, self-hosted AI Metadata tracking tool designed to handle 100,000s of tracked metadata sequences.

Two most famous AI metadata applications are: experiment tracking and prompt engineering.

Aim provides a performant and beautiful UI for exploring and comparing training runs, prompt sessions.

Compare runs easily to build models faster

Compare runs easily to build models faster

  • Group and aggregate 100s of metrics
  • Analyze and learn correlations
  • Query with easy pythonic search
Check out our Github
AimStack

Deep dive into details of each run for easy debugging

Deep dive into details of each run for easy debugging

  • Explore hparams, metrics, images, distributions, audio, text, ...
  • Track plotly and matplotlib plots
  • Analyze system resource usage
Check out our Github
AimStack

Have all relevant information centralized for easy governance

Have all relevant information centralized for easy governance

  • Centralized dashboard to view all your runs
  • Use SDK to query/access tracked runs
  • You own your data - Aim is open source and self hosted.
Check out our Github
AimStack

Demos

Play with Aim before installing it! Check out our demos to see the full functionality

Subscribe to Our Newsletter

Subscribe to our newsletter to receive regular updates about our latest releases, tutorials and blog posts!

Subscribe