Deep Learning Energy Measurement and Optimization

---

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.

Documentation Organization

• Getting Started: Instructions on installation and setup.
• Measuring Energy: How to measure time and energy programmatically and on the command line.
• Optimizing Energy: How to optimize energy.
• Research Overview: Overview of the research papers Zeus is rooted on.
• Source Code Reference: Auto-generated source code reference for the entire codebase.

We also provide usage examples in our GitHub repository.

---

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

Contact

Jae-Won Chung (jwnchung@umich.edu)