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 power-consumption software-systems sustainability
Last synced: about 19 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 (over 2 years ago)
- Default Branch: master
- Last Pushed: 2025-04-22T15:33:33.000Z (5 days ago)
- Last Synced: 2025-04-22T22:01:41.935Z (4 days ago)
- Topics: deep-learning, energy, mlsys
- Language: Python
- Homepage: https://ml.energy/zeus
- Size: 154 MB
- Stars: 245
- Watchers: 9
- Forks: 34
- Open Issues: 20
- Releases: 18
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- Funding: .github/FUNDING.yml
- License: LICENSE
README.md
Project News ⚡
- [2024/08] Perseus, an optimizer for large model training, was accepted to SOSP'24! Paper | Blog | Optimizer
- [2024/07] Added AMD GPU, CPU, and DRAM energy measurement support, and preliminary JAX support!
- [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!
- [2023/07] We used the
ZeusMonitor
to profile GPU time and energy consumption for the ML.ENERGY leaderboard & Colosseum.
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
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 ([email protected])
Owner metadata
- Name: ML.ENERGY
- Login: ml-energy
- Email: [email protected]
- Kind: organization
- Description: Making modern ML energy-efficient
- Website: https://ml.energy
- Location: Ann Arbor, MI
- Twitter:
- Company:
- Icon url: https://avatars.githubusercontent.com/u/109987045?v=4
- Repositories: 9
- Last ynced at: 2025-04-22T21:56:36.705Z
- Profile URL: https://github.com/ml-energy
GitHub Events
Total
- Create event: 28
- Release event: 2
- Issues event: 15
- Watch event: 44
- Delete event: 24
- Member event: 1
- Issue comment event: 35
- Push event: 224
- Pull request review comment event: 234
- Pull request review event: 192
- Pull request event: 59
- Fork event: 8
Last Year
- Create event: 28
- Release event: 2
- Issues event: 15
- Watch event: 44
- Delete event: 24
- Member event: 1
- Issue comment event: 35
- Push event: 224
- Pull request review comment event: 234
- Pull request review event: 192
- Pull request event: 59
- Fork event: 8
Committers metadata
Last synced: 7 days ago
Total Commits: 332
Total Committers: 14
Avg Commits per committer: 23.714
Development Distribution Score (DDS): 0.136
Commits in past year: 127
Committers in past year: 12
Avg Commits per committer in past year: 10.583
Development Distribution Score (DDS) in past year: 0.283
Name | Commits | |
---|---|---|
Jae-Won Chung | j****g@u****u | 287 |
Wonbin Jin | 1****n | 11 |
Parth Raut | 6****t | 8 |
Luoxi Meng | 6****m | 5 |
Sharon Seungyu Han | 8****h | 4 |
danielhou0515 | 7****5 | 3 |
Jisang Ahn | 1****1 | 3 |
Yongseung Lee | 5****1 | 2 |
Ting Sun | s****k@g****m | 2 |
HGangloff | h****f@i****r | 2 |
Divyam Sharma | 4****m | 2 |
Zhenning Yang | z****9@g****m | 1 |
Yu Fan | 4****n | 1 |
Dorota Kopczyk | d****k@g****m | 1 |
Committer domains:
Issue and Pull Request metadata
Last synced: 2 days ago
Total issues: 57
Total pull requests: 114
Average time to close issues: 4 months
Average time to close pull requests: 5 days
Total issue authors: 13
Total pull request authors: 16
Average comments per issue: 1.37
Average comments per pull request: 0.82
Merged pull request: 97
Bot issues: 0
Bot pull requests: 0
Past year issues: 34
Past year pull requests: 100
Past year average time to close issues: about 1 month
Past year average time to close pull requests: 4 days
Past year issue authors: 10
Past year pull request authors: 14
Past year average comments per issue: 1.03
Past year average comments per pull request: 0.67
Past year merged pull request: 84
Past year bot issues: 0
Past year bot pull requests: 0
Top Issue Authors
- jaywonchung (39)
- wbjin (3)
- Rosie-m (2)
- cmcantalupo (2)
- parthraut (2)
- sharonsyh (2)
- michahn01 (1)
- zyang37 (1)
- amirzenoozi (1)
- Sunt-ing (1)
- andife (1)
- FuryMartin (1)
- ImahnShekhzadeh (1)
Top Pull Request Authors
- jaywonchung (55)
- wbjin (14)
- parthraut (10)
- sharonsyh (7)
- michahn01 (6)
- DdIiVvYyAaMm (5)
- danielhou0515 (3)
- show981111 (2)
- Sunt-ing (2)
- vishwa-11 (2)
- jxunn (2)
- HGangloff (2)
- dkopczyk (1)
- fwrrong (1)
- ImahnShekhzadeh (1)
Top Issue Labels
- enhancement (32)
- good first issue (10)
- integration (4)
- documentation (2)
- roadmap (2)
- bug (2)
- cpu (2)
- maintenance (1)
Top Pull Request Labels
Package metadata
- Total packages: 4
-
Total downloads:
- pypi: 2,918 last-month
- cargo: 2,436 total
- Total dependent packages: 1 (may contain duplicates)
- Total dependent repositories: 2 (may contain duplicates)
- Total versions: 27
- 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.11.0 (published 3 months ago)
- Last Synced: 2025-04-25T14:05:02.434Z (2 days ago)
- Versions: 3
- Dependent Packages: 1
- Dependent Repositories: 2
- Downloads: 614 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 3 months ago)
- Last Synced: 2025-04-25T14:05:03.007Z (2 days ago)
- Versions: 20
- Dependent Packages: 0
- Dependent Repositories: 0
- Downloads: 2,304 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 11 months ago)
- Last Synced: 2025-04-25T14:05:03.182Z (2 days ago)
- Versions: 2
- Dependent Packages: 0
- Dependent Repositories: 0
- Downloads: 1,082 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.2.0 (published 3 months ago)
- Last Synced: 2025-04-25T14:05:02.998Z (2 days ago)
- Versions: 2
- Dependent Packages: 0
- Dependent Repositories: 0
- Downloads: 1,354 Total
-
Rankings:
- Dependent repos count: 27.35%
- Dependent packages count: 36.273%
- Average: 53.36%
- Downloads: 96.457%
- Maintainers (1)
Dependencies
- datasets ==2.3.2
- numpy ==1.22.3
- pandas ==1.4.2
- black *
- mkdocs-gen-files ==0.3.5
- mkdocs-literate-nav ==0.4.1
- mkdocs-section-index ==0.3.4
- mkdocstrings ==0.19.0
- 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
- 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
- nvidia/cuda 11.3.1-devel-ubuntu20.04 build
- torch *
- torchvision *
- datasets >=1.8.0
- protobuf *
- scikit-learn *
- scipy *
- sentencepiece *
- torch >=1.3
- transformers ==4.17.0
- torch *
- torchvision *
- torch *
- torchvision *
- numpy *
- nvidia-ml-py *
- pandas *
- pydantic *
- rich *
- scikit-learn *
Score: 16.80531982509633