zeus-ml

View on PyPIReverse Dependencies (1)

0.9.1 zeus_ml-0.9.1-py3-none-any.whl

Wheel Details

Project: zeus-ml
Version: 0.9.1
Filename: zeus_ml-0.9.1-py3-none-any.whl
Download: [link]
Size: 186279
MD5: 58f6522d92e1cf02616a66b8f6067dec
SHA256: 7af196cc4b025efa3ba77a85c6e21946b70cd72d64f2adc77af4bb39b54c737a
Uploaded: 2024-05-07 04:06:47 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: zeus-ml
Version: 0.9.1
Summary: A framework for deep learning energy measurement and optimization.
Author-Email: Jae-Won Chung <jwnchung[at]umich.edu>
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,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
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: pydantic (<2); extra == "bso"
Requires-Dist: httpx; extra == "bso"
Requires-Dist: fastapi[all]; 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: zeus-ml[bso,bso-server,lint,migration,pfo-server,test]; extra == "dev"
Requires-Dist: greenlet; extra == "dev"
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: 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: alembic; extra == "migration"
Requires-Dist: sqlalchemy; extra == "migration"
Requires-Dist: pydantic (<2); extra == "migration"
Requires-Dist: python-dotenv; extra == "migration"
Requires-Dist: pydantic (<2); extra == "pfo"
Requires-Dist: httpx; extra == "pfo"
Requires-Dist: fastapi[all]; extra == "pfo-server"
Requires-Dist: pydantic (<2); extra == "pfo-server"
Requires-Dist: lowtime; extra == "pfo-server"
Requires-Dist: aiofiles; extra == "pfo-server"
Requires-Dist: httpx; extra == "pfo-server"
Requires-Dist: torch; extra == "pfo-server"
Requires-Dist: fastapi[all]; extra == "test"
Requires-Dist: sqlalchemy; extra == "test"
Requires-Dist: pydantic (<2); extra == "test"
Requires-Dist: httpx; 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"
Provides-Extra: bso
Provides-Extra: bso-server
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: lint
Provides-Extra: migration
Provides-Extra: pfo
Provides-Extra: pfo-server
Provides-Extra: test
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 4302 characters]

WHEEL

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

RECORD

Path Digest Size
capriccio/generate.py sha256=-UQUwZzwxC300gaP7Et6fPKYSJZ6t4X1GxGCE9KeLEo 3637
docs/gen_ref_pages.py sha256=fH32REiI9NAP62VfTd6sJ-qCgTGTJI-0iiT48xxYF7Y 1864
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/huggingface/run_clm.py sha256=X2YXrnSJlpZZV74uJS6xSm15cYhV8PpBhIi-acOr3z8 30412
examples/huggingface/run_gemma_sft_qlora.py sha256=SCJiRcwGG4IfQxcrvPSvWmZTvx6cTo2RFoj8hS_c_OA 5662
examples/pipeline_frequency_optimizer/profile_p2p.py sha256=QObx1zOqMrScCESTrbNhremx2vlN5pB1zNMwSgJdEtE 2202
examples/pipeline_frequency_optimizer/run_optimization.py sha256=LBGJIJvu8g7Bmb0uBerlbRYzK38p8y4LtwARp24SqTs 8640
examples/power_limit_optimizer/train_dp.py sha256=dOG3Ee2wIfVV137ay6zKOylzkyZwvZhn6BxypolrgGg 12553
examples/power_limit_optimizer/train_single.py sha256=IGgxFR_35j9HEc5hOxg8BFAAmge7QUPiq5Kep-ZN-MI 11621
examples/research_reproducibility/zeus_nsdi23/run_alibaba.py sha256=rta7THdFpXuhzGpdOXoe9yqxD3wPvB0IL93AIiSrDoI 5369
examples/research_reproducibility/zeus_nsdi23/run_single.py sha256=t1xcwqR-_v6KByoeIC2c_jRO2IUpQMM4ixbShoxmMDw 3901
tests/test_monitor.py sha256=dKaDY4EtfINs88bNRWf1RGO41REKqmWl85QT0BryRDg 17477
tests/optimizer/test_power_limit_optimizer.py sha256=QbuVntJvFO5M0Ypi1j7_pXPpOGovF2DO9ZV7JnyHY8Y 11018
tests/optimizer/batch_size/conftest.py sha256=bhFRvEmSWC9KThWvOWKzJLCi6ClYbbn82mgnh9_0Wn4 2799
tests/optimizer/batch_size/simulate_with_server.py sha256=Cxp3iU1BR3VQMrzKsNlj-yV8c_zZQSfP3fMEBPZbST4 20383
tests/optimizer/batch_size/test_client.py sha256=yO7zxVqOIfUea8rsbpDMaJpHM-8p7nc1ioNjnqXqqic 5546
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=VMB4GhNv7ng7WAcs2stVwkQbaw5QI6feCXrFhmQqmtc 4307
tests/utils/test_env.py sha256=tbU9i4nwkbPQOv9ru0WnPaqkYZTO9T28lVEK7uq3Gds 2715
zeus/__init__.py sha256=Q7DlHnGVzSc3pxnF4mWwCk8Hh1IYEIqW48qQ6_BpGvs 1126
zeus/callback.py sha256=lplRKK-49ETP-MgS5PXNdmBNEOP8MVME2BT1AA_xphs 3455
zeus/exception.py sha256=aLX5l8cx7ZaYkcQML_admoTJWYTuSF9Rp_6DwSB5byI 351
zeus/_legacy/__init__.py sha256=gP0i1CDky2Ku8gQJGQ8Z5RExpix04vOOodLZiY21Ilg 184
zeus/_legacy/job.py sha256=IUzeEPS9eZvkpXv5iRq16plh2ElXcgtFS48NCTAO7tk 5031
zeus/_legacy/simulate.py sha256=d8kwIIU3v_Tem5_39q0s25XGJGCpMVz6XvzYETD7Xgg 36929
zeus/_legacy/policy/__init__.py sha256=O-eAeDvyTLRsev4JYJPspiyrAEIjr_Kp-E21wcwo9a0 1185
zeus/_legacy/policy/interface.py sha256=ySLyOreRkrCsCqIE1ft_-kOFfZo1lF3-9g-9nhNR01Q 3928
zeus/_legacy/policy/mab.py sha256=tr8k0y8dLETwOjrAZwVTqo2K_cyfcustXwsqH7ScbH0 6664
zeus/_legacy/policy/optimizer.py sha256=bplsjsdivWj-6gUhHJ3RzrJbr4bmc-mtLAzM3W4m7-Y 19971
zeus/device/__init__.py sha256=5hXxBnzvug1f3OCbpE4ILFTaGJii7fHimBg4sKg1UIw 4083
zeus/device/exception.py sha256=WnLyiq1E85Yosta0jUQ35ySzOPTreXpCDAsqYoMNQVs 287
zeus/device/gpu.py sha256=J6qdvQ5EVjA4OqYtoSf8fjBmI4JqijnGT1KhYdIa1DI 37607
zeus/monitor/__init__.py sha256=YA_Lfv7K8f1TJhH2RgQGVlp7F-cQQEeelc7-Cv9twTg 910
zeus/monitor/__main__.py sha256=SYMXfh8qyLWIfRsvWzpkJvAIgSjvycK3z29iR5aq_NE 4257
zeus/monitor/energy.py sha256=XQTOyI-gsGPBnOmJ2BofR9DFQfcgaCOCgo9rqx05epM 12742
zeus/monitor/power.py sha256=USywGzMDh2tLtGhbNNDF--uFv4HieVX8k0tmR7ksao8 9786
zeus/optimizer/__init__.py sha256=-ILeYzLJZNWtF6FnYAbdTvmIb45vgPDHmOF_EUyOChg 633
zeus/optimizer/power_limit.py sha256=f0wQ4sWX_pKHJvAx14SQL_U3bSOcyNqIT2lP1ChnNbs 22745
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=eG0Ob7RDJP0YNC2IbvINmM4UkSBkaia7OR4kBVpyYeo 1031
zeus/optimizer/pipeline_frequency/common.py sha256=cN91k0yyJY10RKAVqr-4DlCnok8I5Ociiaqlru8NMYg 11589
zeus/optimizer/pipeline_frequency/frequency_controller.py sha256=AkT56DaGzCnjVbsMHU4YgSq4tHUGbTsO40uIKo1oIUQ 2959
zeus/optimizer/pipeline_frequency/optimizer.py sha256=Vyxgc7fBVigoLYAgrvw00NnZ67noB1boHxEnPIlDnT8 8591
zeus/optimizer/pipeline_frequency/server/__init__.py sha256=3Y7e6Sa-SFnCjdtgPG4vhYLYGsJn-3ClYseoUK7k42M 1087
zeus/optimizer/pipeline_frequency/server/job_manager.py sha256=kLejVHpJkHYYAu59RHpp_jBgrqH6zfeHfzMrzWUBqzg 9466
zeus/optimizer/pipeline_frequency/server/router.py sha256=14yXkDJ1tN004C4CAcvQbgOLukYZ3Tiwq_TzIYXHE0c 3815
zeus/optimizer/pipeline_frequency/server/scheduler.py sha256=XUxkvjpHkdhYzyhQXQasJKARJnb1IKrKRXocGzAKxjY 12170
zeus/utils/__init__.py sha256=ZDFGV8kcIwLkXDTZ3nQwTOUbVCJI9_sf3LOJx6DZi5Q 640
zeus/utils/async_utils.py sha256=PBNkunP0InADDQEEjF_P0HPcAZOIB96ThoDk7AASd70 2024
zeus/utils/env.py sha256=2exKUO4x-jtL270kn-h0BNp-Cw9yQgdr9p4AQQGmGKk 1368
zeus/utils/framework.py sha256=C-QCkucst_F4TwZ-hGgoY41RubdjXU-fFijxuVzf2ZI 1784
zeus/utils/logging.py sha256=Et8RdabXDBgXJ58C_OcWY15o8FUNE6hAMXcQYNEhzqc 1878
zeus/utils/lr_scaler.py sha256=MrgUwxPE8OoBQSwGDkQxyf2voEq43MvecV-FhHRKyFs 1404
zeus/utils/metric.py sha256=a1pdSEiScAAECg2VusjtrSe_LyQQkWla3YlFjpYbSJQ 4391
zeus/utils/pydantic_v1.py sha256=xXhS8KubsJr4G4FDXzMVyB4u1759o5I6yCbgN5BOAvo 1082
zeus/utils/testing.py sha256=6inBxFNMsx0rqEIRoo-HMxgANVY0WnYqO_3VSMVpZC0 6845
zeus_ml-0.9.1.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
zeus_ml-0.9.1.dist-info/METADATA sha256=nBLbHCJ95YMQWe6CWWFijlCZ6TKg4r-DJDzhSyogDZ0 7646
zeus_ml-0.9.1.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
zeus_ml-0.9.1.dist-info/top_level.txt sha256=C9KtWe2ch-jeRTBrLtGuwMMzo-sIbt-rL0SVD8exAQg 35
zeus_ml-0.9.1.dist-info/RECORD

top_level.txt

capriccio
docs
examples
tests
zeus