Catalog PyPI
Back to packageRELEASE DETAILS

0.5.0

PyPI

aibooster

AIBooster - Performance intelligence and observability tools for AI workloads

Published27 Feb 2026
Direct dependencies22
Transitive dependenciesNot calculated

README

Version 0.5.0 documentation

AIBooster

PyPI version Python License

AIBooster is a performance engineering platform for AI workloads.

Overview

AIBooster is an integrated platform that provides performance intelligence and performance observability for AI/ML workloads, consisting of the following components:

  • Performance Intelligence (PI): Performance optimization framework
  • Performance Observability (PO): Dashboards and agents for performance monitoring

Key Features

  • Auto performance tuning
    • Framework-friendly design supporting MMEngine, DeepSpeed, and Megatron-LM
    • Kubernetes support
  • Inference optimization
    • Automatic model conversion for deep learning compilers (TensorRT, etc.)
  • NCCL analysis tools
    • Communication pattern analyzer
    • Benchmark for PyTorch distributed training

Installation

pip install aibooster

Documentation

For detailed usage and API reference, please visit the official documentation.

Examples

For usage examples, please see aibooster-examples.

Downloads & popularity

Awaiting observations

No metric observations yet

Download counts and popularity will appear here as the collector receives them.

Declared dependencies 22

Native requirements
22 direct targetsNot calculated transitive count
Package / targetRequirementRelationship
optuna
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
optuna>=4.4.0runtime
optuna-integration
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
optuna-integration>=4.4.0runtime
scikit-learn
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
scikit-learn>=0.24.2runtime
plotly
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
plotly>=4.9.0runtime
kaleido
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
kaleido<=0.2.1runtime
psutil
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
psutilruntime
nvidia-ml-py
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
nvidia-ml-pyruntime
opentelemetry-api
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
opentelemetry-api==1.30.0runtime
opentelemetry-exporter-otlp
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
opentelemetry-exporter-otlp==1.30.0runtime
opentelemetry-sdk
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
opentelemetry-sdk==1.30.0runtime
kubernetes
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
kubernetes==31.0.0runtime
onnx
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
onnx<1.20.0runtime
onnx_graphsurgeon
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
onnx_graphsurgeonruntime
onnxruntime
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
onnxruntimeruntime
rich
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
richruntime
pandas
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
pandasruntime
onnxscript
Environment
{
  "extras": "",
  "marker": "python_version > \"3.9\"",
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
onnxscript; python_version > "3.9"runtime
packaging
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
packagingruntime
pytest
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
pytest; extra == "dev"runtime
pytest-cov
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
pytest-cov; extra == "dev"runtime
pydoc-markdown
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
pydoc-markdown==4.8.2; extra == "dev"runtime
build
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.8",
  "evaluation": "NOT_EVALUATED"
}
build; extra == "dev"runtime

Distribution files

10 collected
aibooster-0.5.0-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl

bdist_wheel · 500.83 KB · Python cp310

Source file
aibooster-0.5.0-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

bdist_wheel · 442.11 KB · Python cp310

Source file
aibooster-0.5.0-cp311-cp311-manylinux2014_aarch64.manylinux_2_17_aarch64.whl

bdist_wheel · 560.07 KB · Python cp311

Source file
aibooster-0.5.0-cp311-cp311-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

bdist_wheel · 501.77 KB · Python cp311

Source file
aibooster-0.5.0-cp312-cp312-manylinux2014_aarch64.manylinux_2_17_aarch64.whl

bdist_wheel · 542.43 KB · Python cp312

Source file
aibooster-0.5.0-cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

bdist_wheel · 484.03 KB · Python cp312

Source file
aibooster-0.5.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

bdist_wheel · 498.59 KB · Python cp38

Source file
aibooster-0.5.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

bdist_wheel · 445.74 KB · Python cp38

Source file
aibooster-0.5.0-cp39-cp39-manylinux2014_aarch64.manylinux_2_17_aarch64.whl

bdist_wheel · 500.95 KB · Python cp39

Source file
aibooster-0.5.0-cp39-cp39-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

bdist_wheel · 447.16 KB · Python cp39

Source file

Metadata history

Meaningful changes
Current collected state Current

Observed since 21 Oct, 10:02 UTC

No superseded metadata states collected. Unchanged observations do not create new revisions.

Record identity

package_version:10576bc99f52c6fbfe2523da8fbdd5be0f8fcbbeb083232cd3179bc608c17d21

Other collected versions

View all collected versions →