A curated list of open technology projects to sustain a stable climate, energy supply, biodiversity and natural resources.

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!

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

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 ([email protected])


Owner metadata


GitHub Events

Total
Last Year

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 Email 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

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

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

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

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 *

Score: 16.80531982509633