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!
- Host: GitHub
- URL: https://github.com/ml-energy/zeus
- Owner: ml-energy
- License: apache-2.0
- Created: 2022-08-13T21:20:30.000Z (almost 4 years ago)
- Default Branch: master
- Last Pushed: 2026-08-10T02:56:49.000Z (4 days ago)
- Last Synced: 2026-08-10T12:36:10.628Z (3 days ago)
- Topics: deep-learning, energy, mlsys
- Language: Python
- Homepage: https://ml.energy/zeus
- Size: 98.1 MB
- Stars: 371
- Watchers: 9
- Forks: 46
- Open Issues: 16
- Releases: 27
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- Funding: .github/FUNDING.yml
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
- Codeowners: .github/CODEOWNERS
- Security: SECURITY.md
- Governance: GOVERNANCE.md
- Maintainers: MAINTAINERS.md
README.md
Project News ⚡
- [2026/02] Supported by Laude Institute's Slingshots // TWO!
- [2025/12] With NVIDIA, Google, and Meta, we led a NeurIPS 25 tutorial on Energy and Power as First‑Class ML Design Metrics!
- [2025/12] The ML.ENERGY leaderboard got a major upgrade to v3. Read our in-depth technical analysis blog post.
- [2025/09] We shared our experience and design philosophy for The ML.ENERGY Benchmark in our NeurIPS 25 D&B Spotlight paper.
- [2025/05] Zeus now supports CPU, DRAM, AMD GPU, Apple Silicon, and NVIDIA Jetson platform energy measurement!
- [2024/11] Perseus, an optimizer for large model training, appeared at SOSP'24! Paper | Blog | Optimizer
- [2024/05] Zeus is now a PyTorch ecosystem project. Read the PyTorch blog post here!
- [2024/02] Zeus was selected as a 2024 Mozilla Technology Fund awardee!
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.
- Zeus (NSDI 23): Paper | Blog | Slides
- Chase (ICLR Workshop 23): Paper
- Perseus (SOSP 24): Paper | Blog | Slides
- The ML.ENERGY Benchmark (NeurIPS 25 D&B Spotlight): Paper | Repository | Leaderboard
- 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
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
- Name: ML.ENERGY
- Login: ml-energy
- Email: admins@ml.energy
- Kind: organization
- Description: Energy as a First-Class Resource in Modern ML
- Website: https://ml.energy
- Location: Ann Arbor, MI
- Twitter:
- Company:
- Icon url: https://avatars.githubusercontent.com/u/109987045?v=4
- Repositories: 18
- Last ynced at: 2026-08-06T07:41:08.701Z
- Profile URL: https://github.com/ml-energy
GitHub Events
Total
- Release event: 5
- Delete event: 57
- Member event: 1
- Pull request event: 108
- Fork event: 12
- Issues event: 21
- Watch event: 89
- Issue comment event: 64
- Push event: 362
- Pull request review comment event: 292
- Pull request review event: 229
- Create event: 71
Last Year
- Release event: 1
- Delete event: 26
- Pull request event: 28
- Fork event: 2
- Issues event: 1
- Watch event: 21
- Issue comment event: 18
- Push event: 97
- Pull request review comment event: 46
- Pull request review event: 25
- Create event: 31
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 | 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
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
- Total packages: 4
-
Total downloads:
- cargo: 3,742 total
- pypi: 6,126 last-month
- Total dependent packages: 1 (may contain duplicates)
- Total dependent repositories: 2 (may contain duplicates)
- Total versions: 42
- Total maintainers: 2
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
- actions/checkout v3 composite
- actions/setup-python v2 composite
- actions/checkout v3 composite
- actions/setup-python v2 composite
- cpina/github-action-push-to-another-repository v1.5 composite
- actions/checkout v3 composite
- actions/setup-python v2 composite
- pypa/gh-action-pypi-publish release/v1 composite
- 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
- numpy *
- nvidia-ml-py *
- pandas *
- pydantic *
- rich *
- scikit-learn *
- datasets >=1.8.0
- protobuf *
- scikit-learn *
- scipy *
- sentencepiece *
- torch >=1.3
- transformers ==4.17.0
- datasets ==2.3.2
- numpy ==1.22.3
- pandas ==1.4.2
- accelerate >=0.12.0
- datasets >=1.8.0
- evaluate *
- protobuf *
- scikit-learn *
- sentencepiece *
- torch >=1.3
- transformers >=4.37.2
- zeus *
- torch *
- torchvision *
- prom/node-exporter latest
- prom/prometheus latest
- prom/pushgateway latest
- accelerate >=0.12.0
- bitsandbytes *
- datasets >=1.8.0
- nvidia-cusparse-cu11 *
- peft *
- torch >=1.3
- transformers >=4.37.2
- trl *
- zeus *
- 268 dependencies
- amdlib-${TARGETARCH} latest build
- rust 1-slim build
- ubuntu 24.04 build
- accelerate >=0.12.0
- datasets >=1.8.0
- evaluate *
- protobuf *
- scikit-learn *
- sentencepiece *
- torch >=1.3
- transformers >=4.37.2
- zeus *
- torch *
- torchvision *
- 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
- actions/checkout v4 composite
- astral-sh/setup-uv v6 composite
- actions/checkout v4 composite
- katyo/publish-crates v2 composite
- actions/cache v4 composite
- actions/checkout v4 composite
- 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
- 152 dependencies
- jax ==0.4.30
- zeus *
- nvidia/cuda 12.9.2-base-ubuntu24.04 build
Score: 18.200506168364655