AI Runtime
A versatile sidecar enabling metric standardization, model downloading, and management.
Quick Start
Installation
AI Runtime can be installed by pip.
pip install aibrix
Model download
The AI Runtime supports model downloading from the following storage backends:
- HuggingFace
- S3
- TOS
For more details on model downloading, please refer to our Runtime docs.
Integrate with inference engines
The AI Runtime hides various implementation details on the inference engine side, providing a universal method to guide model management, as well as expose inference monitoring metrics.
At present, vLLM engine is supported, and in the future, SGLang and other inference engines will be supported.
For more details on integration with vLLM, please refer to our Runtime docs.
Contributing
We welcome contributions from the community! Check out our contributing guidelines to see how you can make a difference.
Build from source
# This may take several minutes
pip install -e .
Lint, Format and Type Check
Before contributing your code, please run the following commands to ensure that your code passes the tests and linting checks.
# install dependencies
poetry install --no-root --with dev --extras gpu-optimizer
# linting, formatting and type checking
bash ./scripts/format.sh
License
AI Runtime is licensed under the APACHE License.
1 maintainer
Metadata
1 maintainer
Release files for aibrix 0.8.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| aibrix-0.8.0.tar.gz | 449.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| aibrix-0.8.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 1.0 MB
Release files / aibrix-0.8.0.tar.gz
| Download URL | aibrix-0.8.0.tar.gz |
|---|---|
| Size | 449.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
8d293b620431755985bdabbc83836485bbbf676f8e1ddbc1f4eb5ef68f49c80e
|
|
BLAKE2b-256 checksum How to use checksums |
a33ca550b3f91e2b7d564ef6704445743d408f4f675f4cc8218b63a80ac1cff5
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/2.5.1 CPython/3.12.15 Linux/6.17.0-1022-azure
|
Release files / aibrix-0.8.0-py3-none-any.whl
| Download URL | aibrix-0.8.0-py3-none-any.whl |
|---|---|
| Size | 598.6 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
3d2c9e2f5c53ab846c3ca025424a7b767d57723e12aa1885ed3b6616f472353e
|
|
BLAKE2b-256 checksum How to use checksums |
447fd10b557b4f99f46825284556d5fcce6d1356dc1411eb59165f83ef549f96
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/2.5.1 CPython/3.12.15 Linux/6.17.0-1022-azure
|
