mauricekraus / aimos

AimOS 🔍 — An easy-to-use modular observability for AI Systems. Extensible, scalable and modular.

Drop a star to support AimOS ⭐

Join AimOS discord community

Open-source modular observability for AI Systems.

Easily log, connect and observe any parts of your AI Systems from experiments to production to prompts to AI system monitoring.

_{SEAMLESSLY INTEGRATES WITH:}

AimStack offers enterprise support that's beyond core AimOS. Contact via [email protected] e-mail.

---

About • Demos • Default logging apps • Quick Start • Documentation • Examples • Community • Blog •

---

ℹ️ About

AimOS is an open-source operating system for logs. With AimOS you can build, run and combine any kind of logging applications - experiment tracking, production monitoring, AI System (LLM-based) monitoring, usage monitoring etc.

The Logging applications are typically a combination of these components:

• The types and relationships of the data being logged
• The observability UI over the data logged
• Automations over the data logged

AimOS comes installed with a number of default logging apps:

• Base App - a basic generic log exploration and the logging primitives
• AI Experiment Tracking App - log and explore your machine learning experiments. Includes integrations with the majority of leading ML frameworks.
• AI Systems Tracing and Debugging Apps - a combination of variety of apps that log from langchain to llamaindex traces all in one place.

Apart from running the logging apps, AimOS comes with explorers and reports.

• Explorers are advanced logs comparison tools for specific kind of logs - they allow to compare 1000s of sessions of metrics, images, text, audio and other types of data.
• Reports are embedded knowledge-base that operate with the apps and explorers seamlessly to enable capture the knowledge built on top of the logged data from the observations through AimOS apps and explorers.

With the rise of AI Systems and the challenges it brings forward, logging apps are going to be a crucial part of the software.

Our mission is to democratize developer tools for building AI.

Base App

A general observability over anything logged with AimOS.

Visualize all the logs ever logged with AimOS for the given project 🗺️
Base types to log common artifacts such as Images, Audio objects, Figures, Metrics
High-level overview of the logs, the types logged and the respective sessions/ containers
Deep-dive into each type of the log

Experiment Tracking App

Log Metadata Across Your ML Pipeline 💾	Visualize & Compare Metadata via UI 📊
ML experiments and any metadata tracking Integration with popular ML frameworks Easy migration from other experiment trackers	Metadata visualization via AimOS Explorers Grouping and aggregation Querying using Python expressions
Run ML Trainings Effectively ⚡	Organize Your Experiments 🗂️
System info and resource usage tracking Real-time alerting on training progress (upcoming)	Detailed run information for easy debugging Centralized dashboard for holistic view

AI Systems Tracing Apps

Log Inputs, Outputs and Actions of Executions 🤖	Visualize & Compare Executions Steps via UI 🔍
Track all the prompts, generations of LLMs Track all the inputs, outputs of tools Capture chains metadata	Deep dive into single execution steps Compare executions side-by-side

🎬 Demos

Check out live AimOS demos NOW to see it in action.

Tracing LangChain-based chatbot executions View Demo  |  View Code	Tracing LlamaIndex query executions View Demo  |  View Code
Tracking PyTorch-based CNN trainings View Demo  |  View Code

🌍 Default logging apps

AimOS comes pre-installed with a wide variety of apps. Here is the full list:

App Name	Description	Category	Docs	Source
base	Base AimOS app for general observability over anything logged with AimOS. Includes base types to log common artifacts, such as Image, Audio object, Figure, Metric.	Base	docs	source
docs	Use this AimOS app to access AimOS docs.	Docs	-	source
langchain_debugger	Debugger for LangChain that logs LLMs prompts and generations, tools inputs/outputs, and chains metadata.	AI Systems Tracing	docs	source
llamaindex_observer	Debugger and observer for LlamaIndex. Logs metadata like retrieval nodes, queries and responses, embeddings chunks, etc.	AI Systems Tracing	docs	source
experiment_tracker	App for tracking and exploring ML experiments. Integrations with various ML libraries, including Acme, CatBoost, fastai, Hugging Face Transformers, Keras, Keras Tuner, LightGBM, MXNet, Optuna, PaddlePaddle, PyTorch Ignite, SDB3, and XGBoost.	Experiment Tracking	docs	source

🏁 Quick start

Follow the steps below to get started with AimOS.

1. Install AimOS on your training environment

pip3 install aimos

2. Integrate AimOS with your code

from aimstack.base import Run, Metric

# Initialize a new run
run = Run()

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

# Init a metric
metric = Metric(run, name='loss', context={'subset': 'training'})

for i in range(1000):
      metric.track(i, epoch=1)

3. Start AimOS server

aimos server

4. Start AimOS UI

aimos ui

👥 Community

AimOS README badge

Add AimOS badge to your README, if you've enjoyed using AimOS in your work:

[![AimOS](https://img.shields.io/badge/powered%20by-AimOS-%231473E6)](https://github.com/aimhubio/aimos)

Contributing to AimOS

Considering contibuting to AimOS? To get started, please take a moment to read the CONTRIBUTING.md guide.

Join AimOS contributors by submitting your first pull request. Happy coding! 😊

Made with contrib.rocks.

More questions?

1. Open a feature request or report a bug
2. Join Discord community server