Zeus

A Framework for Deep Learning Energy Measurement and Optimization.
https://github.com/ml-energy/zeus

Category: Consumption
Sub Category: Computation and Communication

Keywords

deep-learning energy mlsys

Keywords from Contributors

energy-consumption power-consumption sustainability power software-systems

Last synced: about 3 hours ago
JSON representation

Repository metadata

Measure and optimize the energy consumption of your AI applications!

README.md

Slack workspace
Docker Hub
Homepage
Apache-2.0 License


Project News


Zeus is a library for (1) measuring the energy consumption of Deep Learning workloads and (2) optimizing their energy consumption.

Zeus is part of The ML.ENERGY Initiative.

Repository Organization

zeus/
├── zeus/             # ⚡ Zeus Python package
│  ├── monitor/       #    - Energy and power measurement (programmatic & CLI)
│  ├── optimizer/     #    - Collection of time and energy optimizers
│  ├── device/        #    - Abstraction layer over CPU and GPU devices
│  ├── utils/         #    - Utility functions and classes
│  ├── _legacy/       #    - Legacy code to keep our research papers reproducible
│  ├── metric.py      #    - Prometheus metric export support
│  ├── show_env.py    #    - Installation & device detection verification script
│  └── callback.py    #    - Base class for callbacks during training
├── zeusd             # 🌩️ Zeus daemon
├── docker/           # 🐳 Dockerfiles and Docker Compose files
└── examples/         # 🛠️ Zeus usage examples

Getting Started

Please refer to our Getting Started page.
After that, you might look at

Docker image

We provide a Docker image fully equipped with all dependencies and environments.
Refer to our Docker Hub repository and Dockerfile.

Examples

We provide working examples for integrating and running Zeus in the examples/ directory.

Research

Zeus is rooted on multiple research papers.
Even more research is ongoing, and Zeus will continue to expand and get better at what it's doing.

  1. Zeus (NSDI 23): Paper | Blog | Slides
  2. Chase (ICLR Workshop 23): Paper
  3. Perseus (SOSP 24): Paper | Blog | Slides
  4. The ML.ENERGY Benchmark (NeurIPS 25 D&B Spotlight): Paper | Repository | Leaderboard
  5. Where Do the Joules Go? Diagnosing Inference Energy Consumption: ArXiv | Blog

If you find Zeus relevant to your research, please consider citing:

@inproceedings{zeus-nsdi23,
    title     = {Zeus: Understanding and Optimizing {GPU} Energy Consumption of {DNN} Training},
    author    = {Jie You and Jae-Won Chung and Mosharaf Chowdhury},
    booktitle = {USENIX NSDI},
    year      = {2023}
}

Other Resources

  1. Energy-Efficient Deep Learning with PyTorch and Zeus (PyTorch conference 2023): Recording | Slides

Contact

Jae-Won Chung (jwnchung@umich.edu)

Newsletter

Subscribe to the ML.ENERGY newsletter for the latest news on Zeus and other projects by the ML.ENERGY Initiative.


Owner metadata


GitHub Events

Total
Last Year

Committers metadata

Last synced: 1 day ago

Total Commits: 445
Total Committers: 21
Avg Commits per committer: 21.19
Development Distribution Score (DDS): 0.133

Commits in past year: 84
Committers in past year: 6
Avg Commits per committer in past year: 14.0
Development Distribution Score (DDS) in past year: 0.107

Name Email Commits
Jae-Won Chung j****g@u****u 386
Wonbin Jin 1****n 11
Parth Raut 6****t 8
Jisang Ahn 1****1 6
Luoxi Meng 6****m 5
Sharon Seungyu Han 8****h 4
Arihant Tripathy a****r@l****m 3
danielhou0515 7****5 3
Divyam Sharma 4****m 2
HGangloff h****f@i****r 2
Jeff (Junze) Ma 9****1 2
Ruofan Wu 4****u 2
Ting Sun s****k@g****m 2
Yongseung Lee 5****1 2
Dorota Kopczyk d****k@g****m 1
Kitsios Konstantinos k****t@g****m 1
Sufiyan Ul Rehman 1****n 1
Vishweshwar Ramanakumar 6****1 1
Yu Fan 4****n 1
Zhenning Yang z****9@g****m 1
jxunn 1****n 1

Committer domains:


Issue and Pull Request metadata

Last synced: 1 day ago

Total issues: 57
Total pull requests: 230
Average time to close issues: 5 months
Average time to close pull requests: 4 days
Total issue authors: 14
Total pull request authors: 21
Average comments per issue: 0.95
Average comments per pull request: 0.84
Merged pull request: 193
Bot issues: 0
Bot pull requests: 0

Past year issues: 5
Past year pull requests: 36
Past year average time to close issues: 30 days
Past year average time to close pull requests: 1 day
Past year issue authors: 2
Past year pull request authors: 6
Past year average comments per issue: 1.4
Past year average comments per pull request: 1.39
Past year merged pull request: 30
Past year bot issues: 0
Past year bot pull requests: 0

More stats: https://issues.ecosyste.ms/repositories/lookup?url=https://github.com/ml-energy/zeus

Top Issue Authors

  • jaywonchung (41)
  • parthraut (2)
  • sharonsyh (2)
  • wbjin (2)
  • michahn01 (1)
  • zyang37 (1)
  • amirzenoozi (1)
  • cmcantalupo (1)
  • Sunt-ing (1)
  • kitsiosk (1)
  • andife (1)
  • Weiwei1001 (1)
  • wuqs-net (1)
  • ImahnShekhzadeh (1)

Top Pull Request Authors

  • jaywonchung (121)
  • wbjin (28)
  • parthraut (14)
  • DdIiVvYyAaMm (10)
  • sharonsyh (10)
  • michahn01 (9)
  • danielhou0515 (6)
  • HGangloff (4)
  • vishwa-11 (4)
  • Sunt-ing (3)
  • show981111 (3)
  • jxunn (2)
  • fwrrong (2)
  • FuryMartin (2)
  • kitsiosk (2)

Top Issue Labels

  • enhancement (26)
  • good first issue (10)
  • integration (4)
  • bug (3)
  • roadmap (2)
  • documentation (2)
  • cpu (1)
  • maintenance (1)

Top Pull Request Labels


Package metadata

pypi.org: zeus

A framework for deep learning energy measurement and optimization.

  • Homepage: https://ml.energy/zeus
  • Documentation: https://ml.energy/zeus
  • Licenses: Apache-2.0
  • Latest release: 0.16.0 (published about 1 month ago)
  • Last Synced: 2026-08-12T13:32:53.288Z (1 day ago)
  • Versions: 15
  • Dependent Packages: 1
  • Dependent Repositories: 2
  • Downloads: 3,038 Last month
  • Rankings:
    • Dependent packages count: 4.746%
    • Dependent repos count: 11.571%
    • Average: 12.942%
    • Downloads: 22.508%
  • Maintainers (1)
pypi.org: zeus-ml

A framework for deep learning energy measurement and optimization.

  • Homepage: https://ml.energy/zeus
  • Documentation: https://ml.energy/zeus
  • Licenses: Apache 2.0
  • Latest release: 0.11.0 (published over 1 year ago)
  • Last Synced: 2026-08-12T13:34:05.447Z (1 day ago)
  • Versions: 20
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 3,088 Last month
  • Rankings:
    • Dependent packages count: 6.633%
    • Stargazers count: 8.786%
    • Forks count: 11.1%
    • Average: 16.531%
    • Downloads: 25.525%
    • Dependent repos count: 30.611%
  • Maintainers (1)
crates.io: zeus

Zeus daemon

  • Homepage: https://ml.energy/zeus
  • Documentation: https://docs.rs/zeus/
  • Licenses: Apache-2.0
  • Latest release: 0.1.0 (published about 2 years ago)
  • Last Synced: 2026-08-12T13:32:53.243Z (1 day ago)
  • Versions: 2
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 1,505 Total
  • Rankings:
    • Dependent repos count: 29.318%
    • Dependent packages count: 33.812%
    • Average: 34.288%
    • Downloads: 39.734%
  • Maintainers (1)
crates.io: zeusd

Zeus daemon

  • Homepage: https://ml.energy/zeus
  • Documentation: https://docs.rs/zeusd/
  • Licenses: Apache-2.0
  • Latest release: 0.5.0 (published about 1 month ago)
  • Last Synced: 2026-08-12T13:32:53.003Z (1 day ago)
  • Versions: 5
  • Dependent Packages: 0
  • Dependent Repositories: 0
  • Downloads: 2,237 Total
  • Rankings:
    • Dependent repos count: 27.35%
    • Dependent packages count: 36.273%
    • Average: 53.36%
    • Downloads: 96.457%
  • Maintainers (1)

Dependencies

.github/workflows/check_homepage_build.yaml actions
  • actions/checkout v3 composite
  • actions/setup-python v2 composite
.github/workflows/deploy_homepage.yaml actions
  • actions/checkout v3 composite
  • actions/setup-python v2 composite
  • cpina/github-action-push-to-another-repository v1.5 composite
.github/workflows/publish_pypi.yaml actions
  • actions/checkout v3 composite
  • actions/setup-python v2 composite
  • pypa/gh-action-pypi-publish release/v1 composite
.github/workflows/push_docker.yaml actions
  • actions/checkout v3 composite
  • docker/build-push-action v3 composite
  • docker/login-action v2 composite
  • docker/metadata-action v4 composite
  • docker/setup-buildx-action v2 composite
pyproject.toml pypi
  • numpy *
  • nvidia-ml-py *
  • pandas *
  • pydantic *
  • rich *
  • scikit-learn *
examples/batch_size_optimizer/capriccio/requirements.txt pypi
  • datasets >=1.8.0
  • protobuf *
  • scikit-learn *
  • scipy *
  • sentencepiece *
  • torch >=1.3
  • transformers ==4.17.0
examples/research_reproducibility/zeus_nsdi23/capriccio/requirements.txt pypi
  • datasets ==2.3.2
  • numpy ==1.22.3
  • pandas ==1.4.2
examples/carbon_emission_monitor/requirements.txt pypi
  • accelerate >=0.12.0
  • datasets >=1.8.0
  • evaluate *
  • protobuf *
  • scikit-learn *
  • sentencepiece *
  • torch >=1.3
  • transformers >=4.37.2
  • zeus *
examples/power_limit_optimizer/requirements.txt pypi
  • torch *
  • torchvision *
docker/prometheus/docker-compose.yml docker
  • prom/node-exporter latest
  • prom/prometheus latest
  • prom/pushgateway latest
examples/huggingface/requirements-qlora.txt pypi
  • accelerate >=0.12.0
  • bitsandbytes *
  • datasets >=1.8.0
  • nvidia-cusparse-cu11 *
  • peft *
  • torch >=1.3
  • transformers >=4.37.2
  • trl *
  • zeus *
zeusd/Cargo.lock cargo
  • 268 dependencies
zeusd/packaging/docker/Dockerfile docker
  • amdlib-${TARGETARCH} latest build
  • rust 1-slim build
  • ubuntu 24.04 build
examples/huggingface/requirements.txt pypi
  • accelerate >=0.12.0
  • datasets >=1.8.0
  • evaluate *
  • protobuf *
  • scikit-learn *
  • sentencepiece *
  • torch >=1.3
  • transformers >=4.37.2
  • zeus *
examples/prometheus/requirements.txt pypi
  • torch *
  • torchvision *
.github/workflows/push_docker_zeusd.yaml actions
  • actions/checkout v4 composite
  • actions/download-artifact v4 composite
  • actions/upload-artifact v4 composite
  • docker/build-push-action v6 composite
  • docker/login-action v3 composite
  • docker/metadata-action v5 composite
  • docker/setup-buildx-action v3 composite
.github/workflows/zeus_fmt_lint_test.yaml actions
  • actions/checkout v4 composite
  • astral-sh/setup-uv v6 composite
.github/workflows/publish_crates_io.yaml actions
  • actions/checkout v4 composite
  • katyo/publish-crates v2 composite
.github/workflows/zeusd_fmt_lint_test.yaml actions
  • actions/cache v4 composite
  • actions/checkout v4 composite
zeusd/Cargo.toml cargo
  • reqwest 0.12 development
  • tempfile 3 development
  • actix-web 4
  • anyhow 1
  • clap 4.5
  • futures 0.3
  • humantime 2
  • jsonwebtoken 9
  • libloading 0.8
  • nvml-wrapper 0.12
  • once_cell 1.7
  • paste 1
  • regex 1
  • serde 1
  • serde_json 1
  • shell-words 1
  • thiserror 1
  • tokio 1
  • tokio-stream 0.1
  • toml 0.8
  • tracing 0.1
  • tracing-actix-web 0.7
  • tracing-log 0.2
  • tracing-subscriber 0.3
uv.lock pypi
  • 152 dependencies
examples/jax/requirements.txt pypi
  • jax ==0.4.30
  • zeus *
docker/Dockerfile docker
  • nvidia/cuda 12.9.2-base-ubuntu24.04 build

Score: 18.200506168364655