zeus-ml

View on PyPIReverse Dependencies (1)

0.11.0.post1 zeus_ml-0.11.0.post1-py3-none-any.whl

Wheel Details

Project: zeus-ml
Version: 0.11.0.post1
Filename: zeus_ml-0.11.0.post1-py3-none-any.whl
Download: [link]
Size: 227090
MD5: d8071b914c70b9130d6542d84b3a8f07
SHA256: d0554ff3a4b8a27d818f4ea8688daa500be5f10322fb5824365aef24c1682e43
Uploaded: 2025-02-03 01:17:04 +0000

dist-info

METADATA

Metadata-Version: 2.2
Name: zeus-ml
Version: 0.11.0.post1
Summary: A framework for deep learning energy measurement and optimization.
Author: Zeus Team
Project-Url: Repository, https://github.com/ml-energy/zeus
Project-Url: Homepage, https://ml.energy/zeus
Project-Url: Documentation, https://ml.energy/zeus
License: Apache 2.0
Keywords: deep-learning,power,energy,carbon,sustainability,mlsys
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.9
Requires-Dist: numpy
Requires-Dist: pandas
Requires-Dist: scikit-learn
Requires-Dist: nvidia-ml-py
Requires-Dist: pydantic
Requires-Dist: rich
Requires-Dist: tyro
Requires-Dist: httpx
Requires-Dist: amdsmi
Requires-Dist: python-dateutil
Requires-Dist: pydantic (<2); extra == "pfo"
Requires-Dist: fastapi[standard]; extra == "pfo-server"
Requires-Dist: pydantic (<2); extra == "pfo-server"
Requires-Dist: lowtime; extra == "pfo-server"
Requires-Dist: aiofiles; extra == "pfo-server"
Requires-Dist: torch; extra == "pfo-server"
Requires-Dist: pydantic (<2); extra == "bso"
Requires-Dist: fastapi[standard]; extra == "bso-server"
Requires-Dist: sqlalchemy; extra == "bso-server"
Requires-Dist: pydantic (<2); extra == "bso-server"
Requires-Dist: python-dotenv; extra == "bso-server"
Requires-Dist: alembic; extra == "migration"
Requires-Dist: sqlalchemy; extra == "migration"
Requires-Dist: pydantic (<2); extra == "migration"
Requires-Dist: python-dotenv; extra == "migration"
Requires-Dist: prometheus-client; extra == "prometheus"
Requires-Dist: ruff; extra == "lint"
Requires-Dist: black (==22.6.0); extra == "lint"
Requires-Dist: pyright; extra == "lint"
Requires-Dist: pandas-stubs; extra == "lint"
Requires-Dist: transformers; extra == "lint"
Requires-Dist: fastapi[standard]; extra == "test"
Requires-Dist: sqlalchemy; extra == "test"
Requires-Dist: pydantic (<2); extra == "test"
Requires-Dist: pytest (==7.3.2); extra == "test"
Requires-Dist: pytest-mock (==3.10.0); extra == "test"
Requires-Dist: pytest-xdist (==3.3.1); extra == "test"
Requires-Dist: anyio (==3.7.1); extra == "test"
Requires-Dist: aiosqlite (==0.20.0); extra == "test"
Requires-Dist: numpy (<2); extra == "test"
Requires-Dist: mkdocs-material[imaging] (==9.5.19); extra == "docs"
Requires-Dist: mkdocstrings[python] (==0.25.0); extra == "docs"
Requires-Dist: mkdocs-gen-files (==0.5.0); extra == "docs"
Requires-Dist: mkdocs-literate-nav (==0.6.1); extra == "docs"
Requires-Dist: mkdocs-section-index (==0.3.9); extra == "docs"
Requires-Dist: mkdocs-redirects (==1.2.1); extra == "docs"
Requires-Dist: urllib3 (<2); extra == "docs"
Requires-Dist: black; extra == "docs"
Requires-Dist: zeus-ml[bso,bso-server,lint,migration,pfo-server,prometheus,test]; extra == "dev"
Requires-Dist: greenlet; extra == "dev"
Provides-Extra: pfo
Provides-Extra: pfo-server
Provides-Extra: bso
Provides-Extra: bso-server
Provides-Extra: migration
Provides-Extra: prometheus
Provides-Extra: lint
Provides-Extra: test
Provides-Extra: docs
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 5514 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.8.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
docs/gen_ref_pages.py sha256=AJNzNtNqsLJ5x2P3GpRqHcBSFg8iRqYPGy679-qejrE 1261
examples/batch_size_optimizer/capriccio/train.py sha256=POCyeelKkpYG_5c2vGvTiSeySsSB30twP_UvtGkGEOA 12726
examples/batch_size_optimizer/mnist/train_dp.py sha256=irLiCvIaI1dCwt90RBOksrOMEi1vutSit-38wJ2FgiI 9613
examples/batch_size_optimizer/mnist/train_single_gpu.py sha256=t2m6UPbXDSiU6pGfgEBxenhor9XwbN2v306_PexDDwY 8093
examples/carbon_emission_monitor/run_clm.py sha256=oMS2xrm8ftwITBGZW2hmTRJ6VhgXPM8wcBcS79dGwUM 30934
examples/huggingface/run_clm.py sha256=KqLMlzcTwENDm4lcd6VlKwMWiR0WbSwVj15IWRKFzHA 30424
examples/huggingface/run_gemma_sft_qlora.py sha256=SCJiRcwGG4IfQxcrvPSvWmZTvx6cTo2RFoj8hS_c_OA 5662
examples/jax/measure_energy.py sha256=4GyDJwHkeL8OEQ93juTZLW1QXKQhm-h_yoHxWsZID6A 1070
examples/pipeline_frequency_optimizer/profile_p2p.py sha256=2MtAxs7QGYoNcJ2m5R9DHT_Y9uiV-D-toOrbOOgWge0 2204
examples/pipeline_frequency_optimizer/run_optimization.py sha256=wD5_f45G5YiCZEWmdgumy3YE5SH7XRRXJXcrrZbdPeY 8342
examples/power_limit_optimizer/train_dp.py sha256=nfnG8908KswXHs4FOFWFvecZIEw6dliBzsM-ZjH8sqY 12209
examples/power_limit_optimizer/train_fsdp.py sha256=E8APpo4chmGtr2DXMGXuC_xNoTuvgIzlmBWS8NSGo_0 7585
examples/power_limit_optimizer/train_single.py sha256=IGgxFR_35j9HEc5hOxg8BFAAmge7QUPiq5Kep-ZN-MI 11621
examples/prometheus/train_single.py sha256=S88UOhKkbuaJA8azWGy-tWhNicuJz5_QB_fqjT8D4kU 11961
examples/research_reproducibility/zeus_nsdi23/run_alibaba.py sha256=xV7PYI7AbuK6kjLJ0hWukXsKopkG94M5yhrbuGodQbc 4739
examples/research_reproducibility/zeus_nsdi23/run_single.py sha256=2hXdPt9vLzYLNHjZlYP_U2AUd2rLVOByHg0NLWFuRqw 3286
examples/research_reproducibility/zeus_nsdi23/capriccio/generate.py sha256=nbvrKXl0pKAM8sOd0kQ8yA7-qL7yUME_J6lLJsp_Z8I 3022
tests/test_metric.py sha256=t5w5lFOYm2DajQTR0sod3YgGLk4TVNVQfUdRKE2k0nM 9045
tests/test_monitor.py sha256=u_rf0LNFVczfNJDB-J-UTc2xrHvpbnaxZHMN49GtSf4 20744
tests/device/cpu/test_rapl.py sha256=tV3u1e5Or9FjvokCtObgX218bzU8loJvzmafqh4JFSY 8024
tests/device/cpu/test_socket_detection.py sha256=c52OJ-Mc5W8EGmeqMviWXn26R6YjaxqD3r47yrxVVbA 2126
tests/monitor/test_carbon.py sha256=uW5AHLq48MTfNUwMHztNH-bAakk9tnDsh3h4kZirNJw 22803
tests/optimizer/test_power_limit_optimizer.py sha256=9hykCm1hBw0GoAEC9D98gFwxtAPzzWteJNOtKF44wc4 10558
tests/optimizer/batch_size/conftest.py sha256=bhFRvEmSWC9KThWvOWKzJLCi6ClYbbn82mgnh9_0Wn4 2799
tests/optimizer/batch_size/simulate_with_server.py sha256=xcWEHM2kxgXaNa-wv0erSCjMZ_he7tp5XwTrfIP_vKQ 19780
tests/optimizer/batch_size/test_client.py sha256=XLE8qORcLa_co6ou81qwqv3DjHRSXtkfz9P4nzzbsbI 5550
tests/optimizer/batch_size/test_explorer.py sha256=rqb0nOSKKtoGQWe7Y8PI4UUgRglWg_gZ-NU7XUUm-3Q 6857
tests/optimizer/batch_size/test_server.py sha256=AnwQp6dyWehO8KW_cPRCiqzV2kEDRpDe6XI45knojvw 11531
tests/optimizer/batch_size/test_simulator.py sha256=fBn-sZSNJxFHDXybCHN1ZxwmSTy277UjzGuNO0AqjUs 4495
tests/utils/test_env.py sha256=-CSHITUD9WKgS_N3C0Jka5zvbi_1j2Kn-BO10fDg5wQ 2112
zeus/__init__.py sha256=5nkNG_Jg8GcaaatyNpX71PXe_rpPW1e3128g81fT0V8 781
zeus/callback.py sha256=gNdI_zaUFWLWF9gA6sQCcgLVkG1tDcguzNxq8IIbVSw 2852
zeus/exception.py sha256=aLX5l8cx7ZaYkcQML_admoTJWYTuSF9Rp_6DwSB5byI 351
zeus/metric.py sha256=HIdUm0qlB6xTYyrc5WmPsuXLlHPaTKB2w4-P214BcbQ 21637
zeus/show_env.py sha256=1--u_Z0m4dKtc_ShNF27DoaXNQ12sadE7geHg1CUM0Y 3644
zeus/_legacy/__init__.py sha256=gP0i1CDky2Ku8gQJGQ8Z5RExpix04vOOodLZiY21Ilg 184
zeus/_legacy/job.py sha256=dsPR_qEOCg0q9ctkmGhekHjTz-rZrDfR_W-m4SdCvA8 4428
zeus/_legacy/simulate.py sha256=YtfkCu10I7v63c0uuy973eTUb_kgYnkm0AB_bbEPJ1U 36326
zeus/_legacy/policy/__init__.py sha256=nr1UeIR-VTi9xwfuNo_Pvqe241diqFc37JPrXAPnmd0 582
zeus/_legacy/policy/interface.py sha256=r8xYVBqw_9LSmA2l9wfLReSdFWbw77Q1vs88-XHVebY 3325
zeus/_legacy/policy/mab.py sha256=GJkRwGIch-FUu42iL7sDrauSusXMvXT7qzvHEyKsji8 6061
zeus/_legacy/policy/optimizer.py sha256=LBh1hdfb355rA7ZfnxCoOooTVgmriB9ppUgOR-5Y_ys 19368
zeus/device/__init__.py sha256=pUWNje5ovmzqZuml94CodnQixCZOZkBlARFbb46_eCc 123
zeus/device/common.py sha256=1c9DRzYxtFC0R3fr9DGSTL0-qMyfCJ6ym893ZepPKzk 2222
zeus/device/exception.py sha256=mpOMLmXTwO-wBy6lyTcz68Ol9_oCKzdMl4TFB4YQV9g 712
zeus/device/cpu/__init__.py sha256=VJ2ZHGbBuOBMuKFhHdC1JvlV6sAuKp9QvPOFESLSGRw 1863
zeus/device/cpu/common.py sha256=z1mshpPmsRFyTqoAM5FtZluyJ1gq91JeXUJwqtwwuZY 6048
zeus/device/cpu/rapl.py sha256=PLIZOCy_vPF7-MrnEr349pQ70CbZ-zM250jH5ASWNgo 14966
zeus/device/gpu/__init__.py sha256=vN8lMU4NDyhUUNcm2gJxEtt56wGajdCFcp_f3YBsw94 5464
zeus/device/gpu/amd.py sha256=qEh9xvUb0IZ73GXXXkGS3y4ocLBrKxNyJiuyCNWzA98 16910
zeus/device/gpu/common.py sha256=6dDkkwgQbrV7xBIBNs0gyVKjn79l1mXatoKXatEqsSk 19362
zeus/device/gpu/nvidia.py sha256=2qXdbF2us1WFWgr-eBWffp6iwAkk9ZI60JNONIbJNFA 17984
zeus/monitor/__init__.py sha256=4NKSYmQrCj5oP0OaWnmasyj12bApWu-ETjDV5G-Swds 351
zeus/monitor/__main__.py sha256=wz4GUME-seJwWeLaQzDgY2FNzZKcB9WpUTPqHODRorY 3654
zeus/monitor/carbon.py sha256=JYm43yGS2eV-Ooq4Dg_PD6Q9fi2fqEhEtKxQt_BpuJU 19985
zeus/monitor/energy.py sha256=73AuKhLMeU6VN9g5_jd8F-TdbMjDcj_6ectp9bkY8xQ 20111
zeus/monitor/power.py sha256=0HnyWKGQNyDZs-ETCtdOD6fctYvf2dytnjaV6CatZZE 9902
zeus/optimizer/__init__.py sha256=f7R9Htz2tvj7VIG68WRJYdULHnlad7XhipnTJzfeb3k 30
zeus/optimizer/power_limit.py sha256=StGd20V3OEbu-lQTDB-gqS1uiSeXqs8JAE7ue-4gAF0 23902
zeus/optimizer/batch_size/__init__.py sha256=eEZ28lCmcbwk13c4lCTTgOdz4vz9p_W_t8zVr0Ex4hA 164
zeus/optimizer/batch_size/client.py sha256=0EbmC10G-R15ny1-S8_m5A-vp9PiH3NEQfpHwJIXMUg 7892
zeus/optimizer/batch_size/common.py sha256=Gpi8jOyUt4FjI_7PsHeGLDd8KPIIwMDqwWV3x5wVntM 5712
zeus/optimizer/batch_size/exceptions.py sha256=0gc2mb-w30eaTAZyz71X-Gj6eY8hQedomsTQYfJedO4 691
zeus/optimizer/batch_size/migrations/env.py sha256=VeLpg3lqkHaTeg37ShTdhRhP7EHAkbabA1pPz-1yfs0 2756
zeus/optimizer/batch_size/server/__init__.py sha256=vSTb3iinpRavDttNqbGfJAO2gR3J6GLt_CHDCxV5zoA 2431
zeus/optimizer/batch_size/server/config.py sha256=poGmsKsB9L1CAlqSTea2_9Xjfz2bxxIsdXzE_WrCqSI 1521
zeus/optimizer/batch_size/server/exceptions.py sha256=7RDDts_1JcFB50n1TUMA-x9uVpuiEtAtQTCCIdcxIDw 1553
zeus/optimizer/batch_size/server/explorer.py sha256=o13pNZOBxa8bHrJonKhjRxwazxSE-zJpr-i5zk6R2RE 5497
zeus/optimizer/batch_size/server/mab.py sha256=6la5wEgKBgepRJgdV17hZN06KO1e-HNvyMg6XKMFu9E 11140
zeus/optimizer/batch_size/server/optimizer.py sha256=RzRbRK8zBilwVhgecXZnEJ-i34KKFvItEmTgc988ps0 10989
zeus/optimizer/batch_size/server/router.py sha256=_7kckpeuIrt_6igF1_vHCnaDneujv3SQAjLnp7zmOZw 7127
zeus/optimizer/batch_size/server/batch_size_state/__init__.py sha256=XQoZw1DXwjqeEHmGXFNfc9YbSVlXQ6mqD9kLA0YHrIA 115
zeus/optimizer/batch_size/server/batch_size_state/commands.py sha256=PVSmI1bxLFntjm4c6v_PR1j2vq6iiEe2JY10hN4SCps 4222
zeus/optimizer/batch_size/server/batch_size_state/models.py sha256=98VfeEi4MkqvyLSijEFCl04jSo-ey4chNtBhlIQYHMI 8675
zeus/optimizer/batch_size/server/batch_size_state/repository.py sha256=-kgzwv2ymP-eHpTEIZUXwloEb0JNBqt-mnfv4v3k9eA 9950
zeus/optimizer/batch_size/server/database/__init__.py sha256=C3IsyYvUTUlK_W2BwyNjj3p_uleDoH-XxHGoXVkNQuQ 52
zeus/optimizer/batch_size/server/database/db_connection.py sha256=iPiUJgJ402xNDvDH-RE8yXSYNPvs3WgsykbY5HZkb4A 2600
zeus/optimizer/batch_size/server/database/repository.py sha256=or72S1g04ay86_gcEyfUpABaiWJLRMoY0Nzq9fbIQHQ 411
zeus/optimizer/batch_size/server/database/schema.py sha256=Qik7mAoOz8wBuZEjneH-QnBaj31Ivbq8nc-0zV2cfC8 6931
zeus/optimizer/batch_size/server/job/__init__.py sha256=KwJLfUE7IevYCqmWUk_Es9R8YqYHhhKh4T1Yxp4azCk 55
zeus/optimizer/batch_size/server/job/commands.py sha256=AZEAugo5a-6NnRrclAIAna-OcAV6Vj2op5ScNEyl6pU 5354
zeus/optimizer/batch_size/server/job/models.py sha256=ThIpAud0-8A6elRQyhSENZ3uJb0cFFHv1G4dUKQ3DDQ 2587
zeus/optimizer/batch_size/server/job/repository.py sha256=-CKaUmbFT-sCyQ7vGPPbXz3eMwW5KO9e9DJ4j-Wp_hk 5887
zeus/optimizer/batch_size/server/services/__init__.py sha256=XODPasqyJ9HMMCRsh9M-pPCxAC4pFqV5tlOg14vdyL0 97
zeus/optimizer/batch_size/server/services/commands.py sha256=C0ImOn4biql4qBoEOrncGyMD_hGrplIL-XJfnZupLTU 952
zeus/optimizer/batch_size/server/services/service.py sha256=-cztFaQAVjCFoXWLLzrj_SC3eS-p-3WPqxsMi_TpiIM 14176
zeus/optimizer/pipeline_frequency/__init__.py sha256=Vs2535xgHs32MaUFX-VLrJvGI76TjNMWiwvNulYkB9A 428
zeus/optimizer/pipeline_frequency/common.py sha256=9QpvRaEXMCTOBwEROvil4QPiv8uts45lkDYDY2Gi0p8 10986
zeus/optimizer/pipeline_frequency/frequency_controller.py sha256=pNME9uDHdafkJ93ZyPhrqEvdfSj7pbIiQWjUICkKx18 2383
zeus/optimizer/pipeline_frequency/optimizer.py sha256=VikE8nB3IU9b9wXVBnfd9Ikyd8DX5oM1YBqfCoaVO_M 8019
zeus/optimizer/pipeline_frequency/server/__init__.py sha256=-hGbqN_dhy-dqM6Rmy7t6WBQc3A2vZoVn-RO8OCkFbI 484
zeus/optimizer/pipeline_frequency/server/job_manager.py sha256=XMKl-fFFXESQM-iOvR1cU9wamsRfmVreIIizH1zpZ3Y 8863
zeus/optimizer/pipeline_frequency/server/router.py sha256=-pyPmFYvEEgvb1Tujz-dqjxOxfksgRAnXryTyJmSOQs 3219
zeus/optimizer/pipeline_frequency/server/scheduler.py sha256=4cJfdcGzhSSw-J6CCPTOK8mkeu5ubQ0YGOvzn-4Gm6g 11567
zeus/utils/__init__.py sha256=bPu6CtqDWCRGU7VgW2_aiQrCBi8G89FS6k1PjvDajJ0 37
zeus/utils/async_utils.py sha256=Endz3ZCI_gmjOHe9L0L6CGOw6KzMXleyQsuQqtpdMVw 1421
zeus/utils/env.py sha256=Vk6xeTEd2p9a2jY0pvlHeYj8zfH2kTz-4SsNBVaMfmQ 765
zeus/utils/framework.py sha256=Zc6_TcTxyGBmNauPMVnemSkq_ZcTFBg1Qc89tGc4fCw 5914
zeus/utils/logging.py sha256=JNml2VPwfPlJUuPT6l-HwqrLLzHsijwB1llP1Wz02L0 1275
zeus/utils/lr_scaler.py sha256=-IzTSJ1hwaMPijtPeI2lJSPLCYz4Oi5RJZffvbdNxo8 801
zeus/utils/metric.py sha256=wEoLeaKSzkDoaFJw2OO3_hDGJRr677KYTkugJxXzlVc 3788
zeus/utils/pydantic_v1.py sha256=BmpFw47ImndeZiUCbrox5UTJmA-f8WMZl22q-gdiO0I 479
zeus/utils/testing.py sha256=jU3ouoncFGS6rDm1iJZKJ5vQJwpXONT0VcitkNHdXx4 6286
zeus_ml-0.11.0.post1.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
zeus_ml-0.11.0.post1.dist-info/METADATA sha256=pQGmDwy1tBaYI0pMZMcIACM8mLurTVKHCBIk82SEL4k 8942
zeus_ml-0.11.0.post1.dist-info/WHEEL sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8 91
zeus_ml-0.11.0.post1.dist-info/top_level.txt sha256=wsLaSeMd5gFxPgf-K0C4DOZv0JMdAJivY18kWZ2FUYQ 25
zeus_ml-0.11.0.post1.dist-info/RECORD

top_level.txt

docs
examples
tests
zeus