Skip to content
Zeus logo Zeus logo

An Energy Optimization Framework for DNN Training

Zeus automatically optimizes the energy and time of recurring DNN training jobs by finding the optimal batch size and GPU power limit.

Please refer to our NSDI’23 publication for details. Check out Overview for a summary.

Zeus is part of The ML.ENERGY Initiative.

Getting Started

Refer to Getting Started for instructions on environment setup, installation, and integration. We also provide integration examples:

Extending Zeus

You can easily implement custom policies for batch size and power limit optimization and plug it into Zeus.

Refer to Extending Zeus for details.