Skip to main content

A framework for deep learning energy measurement and optimization.

Project description

Zeus logo

Deep Learning Energy Measurement and Optimization

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
│  └── callback.py    #    - Base class for callbacks during training
│
├── zeusd             # 🌩️ Zeus daemon
│
├── docker/           # 🐳 Dockerfiles and Docker Compose files
│
├── examples/         # 🛠️ Zeus usage examples
│
├── capriccio/        # 🌊 A drifting sentiment analysis dataset
│
└── trace/            # 🗃️ Training and energy traces for various GPUs and DNNs

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 (2023): Paper | Blog | Slides
  2. Chase (2023): Paper
  3. Perseus (2023): Paper | 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

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

Contact

Jae-Won Chung (jwnchung@umich.edu)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

zeus_ml-0.10.1.tar.gz (153.0 kB view details)

Uploaded Source

Built Distribution

zeus_ml-0.10.1-py3-none-any.whl (194.6 kB view details)

Uploaded Python 3

File details

Details for the file zeus_ml-0.10.1.tar.gz.

File metadata

  • Download URL: zeus_ml-0.10.1.tar.gz
  • Upload date:
  • Size: 153.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.12.6

File hashes

Hashes for zeus_ml-0.10.1.tar.gz
Algorithm Hash digest
SHA256 fee3a19ce04058ef39c5bb6112f0e8dbcd41546584003c778c0f74573f6b1c44
MD5 eac6dd61c806dd1a62996c13649e0c46
BLAKE2b-256 22bcc54594e767fe1c40008d568970d4f761fb6a390680be11818a7cf1fe2cd4

See more details on using hashes here.

File details

Details for the file zeus_ml-0.10.1-py3-none-any.whl.

File metadata

  • Download URL: zeus_ml-0.10.1-py3-none-any.whl
  • Upload date:
  • Size: 194.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.12.6

File hashes

Hashes for zeus_ml-0.10.1-py3-none-any.whl
Algorithm Hash digest
SHA256 2732a7df2789b861f2595543518d0c834bc6c75194741f2a1a609a11f4b98869
MD5 a41a96ebe68bb11c0b30f23c7ca953cb
BLAKE2b-256 3a861d23b1ae610a064f19f080852d6ffd94be55532ee630821a80c3133eae9d

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page