zeus

Measure and optimize the energy consumption of your AI applications!

https://github.com/ml-energy/zeus

Science Score: 59.0%

This score indicates how likely this project is to be science-related based on various indicators:

○
CITATION.cff file
✓
codemeta.json file
Found codemeta.json file
✓
.zenodo.json file
Found .zenodo.json file
✓
DOI references
Found 1 DOI reference(s) in README
✓
Academic publication links
Links to: arxiv.org, acm.org
✓
Committers with academic emails
1 of 17 committers (5.9%) from academic institutions
○
Institutional organization owner
○
JOSS paper metadata
○
Scientific vocabulary similarity
Low similarity (10.7%) to scientific vocabulary

Keywords

deep-learning energy mlsys

Last synced: 6 months ago · JSON representation

Repository

Measure and optimize the energy consumption of your AI applications!

Basic Info

Host: GitHub
Owner: ml-energy
License: apache-2.0
Language: Python
Default Branch: master
Homepage: https://ml.energy/zeus
Size: 154 MB

Statistics

Stars: 290
Watchers: 10
Forks: 36
Open Issues: 15
Releases: 20

Topics

deep-learning energy mlsys

Created over 3 years ago · Last pushed 7 months ago

Metadata Files

Readme Contributing Funding License

Deep Learning Energy Measurement and Optimization

[![Slack workspace](https://badgen.net/badge/icon/Join%20workspace/b31b1b?icon=slack&label=Slack)](https://join.slack.com/t/zeus-ml/shared_invite/zt-36fl1m7qa-Ihky6FbfxLtobx40hMj3VA) [![Docker Hub](https://badgen.net/docker/pulls/symbioticlab/zeus?icon=docker&label=Docker%20pulls)](https://hub.docker.com/r/symbioticlab/zeus) [![Homepage](https://custom-icon-badges.demolab.com/badge/Homepage-ml.energy-23d175.svg?logo=home&logoColor=white&logoSource=feather)](https://ml.energy/zeus) [![Apache-2.0 License](https://custom-icon-badges.herokuapp.com/github/license/ml-energy/zeus?logo=law)](/LICENSE)

Project News ⚡

[2025/05] We shared our experience and design philosophy for the ML.ENERGY leaderboard in this 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: Paper

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

bibtex @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

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

Contact

Jae-Won Chung (jwnchung@umich.edu)

Owner

Name: ML.ENERGY
Login: ml-energy
Kind: organization
Email: admins@ml.energy
Location: Ann Arbor, MI

Website: https://ml.energy
Repositories: 10
Profile: https://github.com/ml-energy

Making modern ML energy-efficient

GitHub Events

Total

Create event: 40
Release event: 4
Issues event: 20
Watch event: 75
Delete event: 31
Member event: 1
Issue comment event: 46
Push event: 265
Pull request review comment event: 246
Pull request review event: 204
Pull request event: 80
Fork event: 10

Last Year

Create event: 40
Release event: 4
Issues event: 20
Watch event: 75
Delete event: 31
Member event: 1
Issue comment event: 46
Push event: 265
Pull request review comment event: 246
Pull request review event: 204
Pull request event: 80
Fork event: 10

Committers

Last synced: 7 months ago

All Time

Total Commits: 361
Total Committers: 17
Avg Commits per committer: 21.235
Development Distribution Score (DDS): 0.139

Past Year

Commits: 93
Committers: 11
Avg Commits per committer: 8.455
Development Distribution Score (DDS): 0.301

Top Committers

Name	Email	Commits
Jae-Won Chung	j**g@u**u	311
Wonbin Jin	1****n	11
Parth Raut	6****t	8
Jisang Ahn	1****1	5
Luoxi Meng	6****m	5
Sharon Seungyu Han	8****h	4
danielhou0515	7****5	3
Divyam Sharma	4****m	2
HGangloff	h**f@i**r	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
Vishweshwar Ramanakumar	6****1	1
Yu Fan	4****n	1
Zhenning Yang	z**9@g**m	1
jxunn	1****n	1

Committer Domains (Top 20 + Academic)

inrae.fr: 1 umich.edu: 1

Issues and Pull Requests

Last synced: 6 months ago

All Time

Total issues: 51
Total pull requests: 129
Average time to close issues: 4 months
Average time to close pull requests: 6 days
Total issue authors: 14
Total pull request authors: 17
Average comments per issue: 1.08
Average comments per pull request: 0.78
Merged pull requests: 111
Bot issues: 0
Bot pull requests: 0

Past Year

Issues: 16
Pull requests: 76
Average time to close issues: about 1 month
Average time to close pull requests: 7 days
Issue authors: 7
Pull request authors: 11
Average comments per issue: 1.13
Average comments per pull request: 0.62
Merged pull requests: 63
Bot issues: 0
Bot pull requests: 0

View more stats

Top Authors

Issue Authors

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

Pull Request Authors

jaywonchung (93)
wbjin (26)
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 Labels

Issue Labels

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

Pull Request Labels

Packages

Total packages: 4
Total downloads:
- pypi 996 last-month
- cargo 3,266 total

Total dependent packages: 1
(may contain duplicates)
Total dependent repositories: 2
(may contain duplicates)
Total versions: 31
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
License: apache-2.0
Latest release: 0.12.1
published 7 months ago

Versions: 7
Dependent Packages: 1
Dependent Repositories: 2
Downloads: 414 Last month

Rankings

Dependent packages count: 4.7%

Dependent repos count: 11.6%

Average: 12.9%

Downloads: 22.5%

Maintainers (1)

jaywonchung

Last synced: 6 months ago

pypi.org: zeus-ml

A framework for deep learning energy measurement and optimization.

Homepage: https://ml.energy/zeus
Documentation: https://ml.energy/zeus
License: Apache 2.0
Latest release: 0.11.0
published about 1 year ago

Versions: 20
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 582 Last month

Rankings

Dependent packages count: 6.6%

Stargazers count: 8.8%

Forks count: 11.1%

Average: 16.5%

Downloads: 25.5%

Dependent repos count: 30.6%

Maintainers (1)

jaywonchung

Last synced: 6 months ago

crates.io: zeus

Zeus daemon

Homepage: https://ml.energy/zeus
Documentation: https://docs.rs/zeus/
License: Apache-2.0
Latest release: 0.1.0
published over 1 year ago

Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 1,366 Total

Rankings

Dependent repos count: 29.3%

Dependent packages count: 33.8%

Average: 34.3%

Downloads: 39.7%

Maintainers (1)

jaywonchung

Last synced: 6 months ago

crates.io: zeusd

Zeus daemon

Homepage: https://ml.energy/zeus
Documentation: https://docs.rs/zeusd/
License: Apache-2.0
Latest release: 0.2.0
published about 1 year ago

Versions: 2
Dependent Packages: 0
Dependent Repositories: 0
Downloads: 1,900 Total

Rankings

Dependent repos count: 27.3%

Dependent packages count: 36.3%

Average: 53.4%

Downloads: 96.5%

Maintainers (1)

jaywonchung

Last synced: 6 months ago

Dependencies

capriccio/requirements.txt pypi

datasets ==2.3.2
numpy ==1.22.3
pandas ==1.4.2

docs/requirements.txt pypi

black *
mkdocs-gen-files ==0.3.5
mkdocs-literate-nav ==0.4.1
mkdocs-section-index ==0.3.4
mkdocstrings ==0.19.0

.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/lint.yaml actions

actions/checkout v3 composite
actions/setup-python v2 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

Dockerfile docker

nvidia/cuda 11.3.1-devel-ubuntu20.04 build

examples/imagenet/requirements.txt pypi

torch *
torchvision *

examples/ZeusDataLoader/capriccio/requirements.txt pypi

datasets >=1.8.0
protobuf *
scikit-learn *
scipy *
sentencepiece *
torch >=1.3
transformers ==4.17.0

examples/ZeusDataLoader/cifar100/requirements.txt pypi

torch *
torchvision *

examples/ZeusDataLoader/imagenet/requirements.txt pypi

torch *
torchvision *

pyproject.toml pypi

numpy *
nvidia-ml-py *
pandas *
pydantic *
rich *
scikit-learn *

zeus

Science Score: 59.0%

Keywords

Repository

Basic Info

Statistics

Topics

Metadata Files

README.md

Deep Learning Energy Measurement and Optimization

- [2024/02] Zeus was selected as a 2024 Mozilla Technology Fund awardee!

Repository Organization

Getting Started

Docker image

Examples

Research

Other Resources

Contact

Owner

GitHub Events

Total

Last Year

Committers

All Time

Past Year

Top Committers

Committer Domains (Top 20 + Academic)

Issues and Pull Requests

All Time

Past Year

Top Authors

Issue Authors

Pull Request Authors

Top Labels

Issue Labels

Pull Request Labels

Packages

pypi.org: zeus

Rankings

Maintainers (1)

pypi.org: zeus-ml

Rankings

Maintainers (1)

crates.io: zeus

Rankings

Maintainers (1)

crates.io: zeusd

Rankings

Maintainers (1)

Dependencies