LICENSE
README.md
pyproject.toml
capriccio/generate.py
docs/gen_ref_pages.py
examples/batch_size_optimizer/capriccio/train.py
examples/batch_size_optimizer/mnist/train_dp.py
examples/batch_size_optimizer/mnist/train_single_gpu.py
examples/huggingface/run_clm.py
examples/huggingface/run_gemma_sft_qlora.py
examples/jax/measure_energy.py
examples/pipeline_frequency_optimizer/profile_p2p.py
examples/pipeline_frequency_optimizer/run_optimization.py
examples/power_limit_optimizer/train_dp.py
examples/power_limit_optimizer/train_single.py
examples/research_reproducibility/zeus_nsdi23/run_alibaba.py
examples/research_reproducibility/zeus_nsdi23/run_single.py
tests/test_monitor.py
tests/device/cpu/test_rapl.py
tests/device/cpu/test_socket_detection.py
tests/optimizer/test_power_limit_optimizer.py
tests/optimizer/batch_size/conftest.py
tests/optimizer/batch_size/simulate_with_server.py
tests/optimizer/batch_size/test_client.py
tests/optimizer/batch_size/test_explorer.py
tests/optimizer/batch_size/test_server.py
tests/optimizer/batch_size/test_simulator.py
tests/utils/test_env.py
zeus/__init__.py
zeus/callback.py
zeus/exception.py
zeus/show_env.py
zeus/_legacy/__init__.py
zeus/_legacy/job.py
zeus/_legacy/simulate.py
zeus/_legacy/policy/__init__.py
zeus/_legacy/policy/interface.py
zeus/_legacy/policy/mab.py
zeus/_legacy/policy/optimizer.py
zeus/device/__init__.py
zeus/device/common.py
zeus/device/exception.py
zeus/device/cpu/__init__.py
zeus/device/cpu/common.py
zeus/device/cpu/rapl.py
zeus/device/gpu/__init__.py
zeus/device/gpu/amd.py
zeus/device/gpu/common.py
zeus/device/gpu/nvidia.py
zeus/monitor/__init__.py
zeus/monitor/__main__.py
zeus/monitor/energy.py
zeus/monitor/power.py
zeus/optimizer/__init__.py
zeus/optimizer/power_limit.py
zeus/optimizer/batch_size/__init__.py
zeus/optimizer/batch_size/client.py
zeus/optimizer/batch_size/common.py
zeus/optimizer/batch_size/exceptions.py
zeus/optimizer/batch_size/migrations/env.py
zeus/optimizer/batch_size/server/__init__.py
zeus/optimizer/batch_size/server/config.py
zeus/optimizer/batch_size/server/exceptions.py
zeus/optimizer/batch_size/server/explorer.py
zeus/optimizer/batch_size/server/mab.py
zeus/optimizer/batch_size/server/optimizer.py
zeus/optimizer/batch_size/server/router.py
zeus/optimizer/batch_size/server/batch_size_state/__init__.py
zeus/optimizer/batch_size/server/batch_size_state/commands.py
zeus/optimizer/batch_size/server/batch_size_state/models.py
zeus/optimizer/batch_size/server/batch_size_state/repository.py
zeus/optimizer/batch_size/server/database/__init__.py
zeus/optimizer/batch_size/server/database/db_connection.py
zeus/optimizer/batch_size/server/database/repository.py
zeus/optimizer/batch_size/server/database/schema.py
zeus/optimizer/batch_size/server/job/__init__.py
zeus/optimizer/batch_size/server/job/commands.py
zeus/optimizer/batch_size/server/job/models.py
zeus/optimizer/batch_size/server/job/repository.py
zeus/optimizer/batch_size/server/services/__init__.py
zeus/optimizer/batch_size/server/services/commands.py
zeus/optimizer/batch_size/server/services/service.py
zeus/optimizer/pipeline_frequency/__init__.py
zeus/optimizer/pipeline_frequency/common.py
zeus/optimizer/pipeline_frequency/frequency_controller.py
zeus/optimizer/pipeline_frequency/optimizer.py
zeus/optimizer/pipeline_frequency/server/__init__.py
zeus/optimizer/pipeline_frequency/server/job_manager.py
zeus/optimizer/pipeline_frequency/server/router.py
zeus/optimizer/pipeline_frequency/server/scheduler.py
zeus/utils/__init__.py
zeus/utils/async_utils.py
zeus/utils/env.py
zeus/utils/framework.py
zeus/utils/logging.py
zeus/utils/lr_scaler.py
zeus/utils/metric.py
zeus/utils/pydantic_v1.py
zeus/utils/testing.py
zeus_ml.egg-info/PKG-INFO
zeus_ml.egg-info/SOURCES.txt
zeus_ml.egg-info/dependency_links.txt
zeus_ml.egg-info/requires.txt
zeus_ml.egg-info/top_level.txt