Repolode
aimhubio

aimhubio/aim

开源、可自托管的实验跟踪工具,用于记录机器学习训练过程与元数据,并提供对比界面和 Python SDK。

可替代:Weights & BiasesTensorBoardMLflow

Star
6,289
Fork
413
未关闭 Issue
483
最近提交
20小时前

这是什么?

Aim 是一款可自托管的机器学习实验跟踪工具。安装 Python 包后,只需在训练代码中添加少量调用,即可记录参数、指标、图片、文本等元数据,再通过 `aim up` 打开网页界面浏览和对比各次训练。

界面支持按已记录的参数对训练运行进行分组、聚合和筛选,查询使用 Python 风格的表达式。项目设计目标是处理上万次训练运行。Python SDK 可以以编程方式读取同一批数据,适合自动化任务和 Jupyter 笔记本分析。内置转换器可导入 TensorBoard、MLflow、Weights & Biases 的日志,并为 PyTorch Lightning、Hugging Face、Keras、XGBoost 等框架提供回调集成。

多主机场景下,可部署集中式远程跟踪服务器汇总各机器的运行记录,同时提供官方 Docker 镜像和 Kubernetes 部署指南。

核心功能

  • 01记录训练运行的参数、指标、图片、音频等元数据。
  • 02网页界面可按跟踪参数对运行进行对比、分组和聚合。
  • 03支持使用 Python 表达式筛选运行和指标。
  • 04提供 Python SDK,便于编程读取数据,也可在 Jupyter 中使用。
  • 05可导入 TensorBoard、MLflow、Weights & Biases 的日志。
  • 06支持集中式远程跟踪服务器,并提供 Docker 与 Kubernetes 部署方式。

适用场景

  • 在同一个面板中对比数百甚至上千次超参数搜索的运行结果。
  • 将已有的 TensorBoard 或 MLflow 日志迁移到可扩展性更好的对比界面。
  • 把多台机器上的训练记录汇总到一个共享的跟踪服务器。
  • 在 Jupyter 笔记本中以编程方式分析实验结果。

使用前须知

  • README 中标注的平台支持仅为 Linux 和 macOS。

收录于专题

标签

常见问题

aim 是什么?
开源、可自托管的实验跟踪工具,用于记录机器学习训练过程与元数据,并提供对比界面和 Python SDK。 Aim 是一款可自托管的机器学习实验跟踪工具。安装 Python 包后,只需在训练代码中添加少量调用,即可记录参数、指标、图片、文本等元数据,再通过 `aim up` 打开网页界面浏览和对比各次训练。
aim 可以免费使用吗?用的什么许可证?
aim 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
aim 还在维护吗?
活跃维护中。aim 最近一次提交是在 2026年10月21日。
aim 适合用来做什么?
在同一个面板中对比数百甚至上千次超参数搜索的运行结果。 将已有的 TensorBoard 或 MLflow 日志迁移到可扩展性更好的对比界面。 把多台机器上的训练记录汇总到一个共享的跟踪服务器。 在 Jupyter 笔记本中以编程方式分析实验结果。
在哪里获取 aim?
源代码和安装说明在 GitHub 上:https://github.com/aimhubio/aim

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目