Wheelodex » Projects » paxml

paxml

View on PyPI — Reverse Dependencies (1)

1.4.0 paxml-1.4.0-py3-none-any.whl

Wheel Details

Project: paxml
Version: 1.4.0
Filename: paxml-1.4.0-py3-none-any.whl
Download: [link]
Size: 440198
MD5: 089c6dbe80b305fcb8a5e62e94295f29
SHA256: 1ed1d84a13a35fa7547f9ba6100f036361f379e54e16bc28e40bb854ca01cb78
Uploaded: 2024-04-09 22:28:59 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: paxml
Version: 1.4.0
Summary: Framework to configure and run machine learning experiments on top of Jax.
Author: PAX team
Author-Email: pax-dev[at]google.com
Home-Page: https://github.com/google/paxml
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10
Requires-Dist: absl-py (==1.4.0)
Requires-Dist: clu (==0.0.11)
Requires-Dist: etils (==1.7.0)
Requires-Dist: flax (==0.8.2)
Requires-Dist: graphviz (==0.20.1)
Requires-Dist: jax (==0.4.26)
Requires-Dist: lingvo (==0.12.7)
Requires-Dist: numpy (==1.26.4)
Requires-Dist: orbax-checkpoint (==0.5.9)
Requires-Dist: praxis (==1.4.0)
Requires-Dist: protobuf (==3.19.6)
Requires-Dist: pyglove (==0.4.4)
Requires-Dist: seqio-nightly (==0.0.17.dev20231010)
Requires-Dist: t5 (==0.9.4)
Requires-Dist: tensorflow-datasets (==4.8.3)
Requires-Dist: tensorflow-metadata (==1.12.0)
Requires-Dist: tensorflow-text (~=2.9.0)
Requires-Dist: tensorflow (~=2.9.2)
Requires-Dist: tensorstore (==0.1.55)
Requires-Dist: tfds-nightly (==4.8.3.dev202303280045)
Requires-Dist: jsonlines (==3.1.0); extra == "gpu"
Requires-Dist: pysimdjson (==5.0.2); extra == "gpu"
Requires-Dist: zstandard (==0.18.0); extra == "gpu"
Provides-Extra: gpu
License-File: LICENSE
[No description]

WHEEL

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

RECORD

Path Digest Size
paxml/automl.py sha256=V2el2kPP4whiIxTSGk5bMlL7tq0-X3yB1f0Edl8qSMI 29105
paxml/automl_interfaces.py sha256=Cgo7mSPETrlVmC-mpEYRREFPs6c6lPM6wT0LRPxBpWE 17158
paxml/automl_test.py sha256=uwPGdCP-qu2Cpfs2ssxtkoG4xS3X7LGsnbHutqHCWEk 36854
paxml/base_executor.py sha256=-y53J8DHpDK9bexYurJ-Zsx8bhovCTcxkUetNSadKi4 2030
paxml/base_experiment.py sha256=wHykFf_SDSYe4P0v9dnMhCRpOyiexWgo5gJVw_tvpB4 7176
paxml/base_inference_runner.py sha256=wva_A7SYoOEzYI3Vxr1JQ0jdNKPiDCvHALdEbKOy5XQ 3214
paxml/base_inference_runner_test.py sha256=ZqM26D0VW-YTQRf8YtsI-0q7ZJXQFsj7gNVxeJEB0Jg 3086
paxml/base_metrics.py sha256=EBCmzDYyOu8H2Sd9jmeuslzUp0vCJPOucJjtc5E9aRk 17258
paxml/base_metrics_test.py sha256=3cOSHCvb-HTVXuwECpK8chAyo0wBax4zKZXYR3mCSes 7160
paxml/base_task.py sha256=LpsiwBqM-MoGPc2ZfTR4IcP6yrhmTGaD4G6NEx58QfA 1106
paxml/checkpoint_creators.py sha256=sFkX9fBCSAG1oETDdN0pbHkJ54HX6J-H1MsefSR5vD0 22776
paxml/checkpoint_managers.py sha256=8KAQ94_IHK9XRMVLFIBSMPAJQD_zlB9Ss6eEr8FYSQw 21268
paxml/checkpoint_managers_test.py sha256=8UV6tZwDXPLkqGVBcyFMGS46xNQ-qMg3SPpuC5NI7S4 36573
paxml/checkpoint_metadata.py sha256=QCF0ORIpieF-aqIzBnm1cAI_vs_b0lNG7y1vDNyvtuM 9996
paxml/checkpoint_paths.py sha256=YIYISY7LyB4ZVu7cQlyfmOstxa9brVrObke5BmT6QY4 11335
paxml/checkpoint_types.py sha256=g-4L9sqyWPkVWe_gnJiodfeUJpKQRV0jek6s2YBhg1Q 1555
paxml/checkpoint_version.py sha256=wYWgc2A0yiBEfDSQcmjqT_pzzm3Ajy2jL19tHHo_gAk 2030
paxml/checkpoints.py sha256=ZaCitimyKMn7E-ZTE-kD0jiuykDXxF7iMZNvfD7CiNI 30794
paxml/checkpoints_test.py sha256=rUSSIsNrMIXN_KUe1c7S36X6kNMpQc5ADCXwWpnnNUQ 13204
paxml/decode_programs.py sha256=ssT_0RbbIl78T0ZOyGTTcBaoeT0E3U-S8xruw8lYOko 16889
paxml/eval_lib.py sha256=O1IOyN91wgqPN4xtC6EPPabit1zusoa8mc65h_bP610 39112
paxml/executors.py sha256=kcut8SzS33Tl2sE48jlat21dFvNECbw594waTL7y1XE 20594
paxml/experiment_imports_all_test.py sha256=Xhd1LIbVZICjnH4jHM-9wQxy1csNKTaCueOE_P3nSTA 1659
paxml/experiment_imports_test_helper.py sha256=tWAkqHAx80IgrQ_qHTJqfeD27ZFdgrfosw7hysu9j5k 5149
paxml/experiment_registry.py sha256=YCDm5snnn70k0prechpDfPJkxdu9rc2B4Gw5PB0j830 6734
paxml/experiment_registry_test.py sha256=Hpek1PaoI-VvfYMp2UB2h_ycs5-bIPQkTMvGgm_rn84 4771
paxml/experiment_utils.py sha256=VCsn0Slv6AhbDnNjnifGHX8VDaYj0NtUFOVa2G9IEUg 4202
paxml/first_result_metric_callback.py sha256=Z3WqvdgKq8vmuGFDuqBbByAVaTn7JwXXY9m_Z8i4ZXY 651
paxml/host_callback.py sha256=FxHFYJb2CB_97ex55p6HVkxCOL90j8WFi_9corG7uVM 4496
paxml/host_callback_test.py sha256=l3GvzPb2Mmp0w-TlqaffUbIR2puQW0747ra9S-Gb89o 2683
paxml/io_utils.py sha256=cHuET5SLiP-NhWeyWUza353Q2Jpq4cjDzse0wr6KA48 13232
paxml/io_utils_test.py sha256=STvq3eYOKzZemuMyTVo5VKc5XY_zVlGYpd2mYcRlEWk 6381
paxml/learners.py sha256=auAn1AZyDu1-Yj-WOymlnPIhCIavee4VFDMe71mhZr0 26711
paxml/learners_test.py sha256=I7u60yjoNdMS11s5TqwYv50Bu2kz6kQ6zc63wZVnU5g 46163
paxml/main.py sha256=cgjRVuT9yTozydPK6P9JqTrPVdK_xQHsYzm5D5v7IPo 22105
paxml/main_test.py sha256=xjCpa7X5pMYp-LaDF_dQQ1GkJIBoAiEaP1aP1Guv-wU 2207
paxml/metric_utils.py sha256=PnpmlBopaS1Zv7g2mbW45ozmCC1oyBQO8DGC1NtSosM 7549
paxml/metric_utils_test.py sha256=rPX2TJhrsSxMkDPzkF2pUbwFe1ABbUXvG1vN9vs3rm8 13059
paxml/parameterized_experiment.py sha256=kZ9jn34sOFPkpZ3hnedwKOg4a3XtYt341Yy8mq3Fh4Q 7770
paxml/parameterized_experiment_test.py sha256=8e9i-hYEgC2HyTIhih__zZoxUunlxnp88bBu3GeITm4 6055
paxml/partitioning.py sha256=jPbq8lX_45ya8N3_UbFzy0rwoAewrKlvIJCb_PXlT2c 58214
paxml/partitioning_test.py sha256=ZYP9PDJYyELqvWeeM5QkZVWjDGY-UDhcDQ-g1xpV6xE 1863
paxml/profiling.py sha256=rVeW9arFeiaagbG_S9sDVn-RS32dDOTu4HqumTJpdik 3130
paxml/programs.py sha256=TdzCLQ-mF-kwc9t5_R-QvuM2CHaCLjODU-L-bpIT3eU 37241
paxml/programs_test.py sha256=QVztTKfZrJpvIJrTYQObo5J5XDOq0mARvVAsKpDRAOw 4557
paxml/seqio_input.py sha256=Yb8myCrpREm1kwzFfa2SPWImFXagKNa3c8KcJE7cnWg 77003
paxml/seqio_input_test.py sha256=cKdNn9_Rg1Jr2LyanY2M3w2jlijAQkkC_2OKw39pHC0 53698
paxml/setup_jax.py sha256=e_M0U78cqKGeBAhqocGK1d9TBoEzsdOUgK7_lx1vk-U 2830
paxml/sgf.py sha256=uiAy4lA2opyRz4dF70xoy7aOWTq02lcxYuWrWM46EcM 29994
paxml/summary_utils.py sha256=bQForSPJeDppSbaF1RvFVbiLkfwiQzVdUY5SwAM7gKg 34279
paxml/summary_utils_test.py sha256=4FG_W0HjklO3RCkxAdTuLmZx3wzTt9utFc8zA-PV9HI 17018
paxml/tasks_lib.py sha256=arx1V-Ven8IXKFH3OtCSFnmvGYNYI7devyXLYe89rRY 79592
paxml/tasks_lib_test.py sha256=GsTpYdRROZWMZxwQHGiXtC_AEBCIaBB4U7AA0Art81k 54788
paxml/test_helper.py sha256=KBP7ihH-tgd60iKsaZdVJ7TOjGNsqXBlNAY570jDhw0 793
paxml/tf_data_service_lib.py sha256=8bTkh6oCiHChlZHDYCnu1CWo2SSqzC4oCdacfqooRYY 5670
paxml/train.py sha256=vOWD2dG7pfI44gLUvaM8tmXlM-jdui7pAJXzY50lMBc 11267
paxml/train_states.py sha256=tvOR8F_byBYEc8Qyre5Xl-O8pNM3PHP2pV3ZOCQq0zs 4136
paxml/trainer_lib.py sha256=BXwYYIVsu32mF34aiW8zcNgY5qgG7uz7o4PsxMCp2fc 61465
paxml/trainer_lib_test.py sha256=70oIQQCwD6UyZEQLySuCd_jTjQTWEVDu8fpSx-fJ6gc 7249
paxml/tuning_lib.py sha256=N153juwR8iDR5YP-ruSEDVtERtT0LbzEHd93485tZgY 39966
paxml/tuning_lib_test.py sha256=Q32ZhpxQHTu62cbDGu7qL8nXqzg8g8L_iP_0xi0eaKg 34450
paxml/xla_passthrough.py sha256=2I49n5GnOxSQ2bmYJ8jkgtIgizM1attCKW0tMXbGhi8 3864
paxml/xla_passthrough_test.py sha256=3AMDl3ltdnhpMvxoigO-Z6GTvxZv6xBQvaCbDM2kgWU 4354
paxml/contrib/gpu/scripts_gpu/configs.py sha256=MEj3SCgatwqzv66icoWrj535AtRPCh-pxTM_WzQbIlg 11498
paxml/contrib/gpu/scripts_gpu/download_boolq.py sha256=dvDn2UkTo9owuB3122xyoELkhHDSv0cv-yVSjqr7_oA 748
paxml/contrib/gpu/scripts_gpu/download_lambada.py sha256=0JhQcW3YM8BPOX-Iu4WRtr-MYzNWCj8RZQVTzzBhG9A 784
paxml/contrib/gpu/scripts_gpu/download_the_pile.py sha256=r7KDILgYl2P210QzZ57yx88uG6bmPa0lrFDyseCRZqo 777
paxml/contrib/gpu/scripts_gpu/llama_utils.py sha256=fGsw5MDmbfsqlm8TLMHz2eDgwej7pPTrsW8rKXyFFJg 6384
paxml/contrib/gpu/scripts_gpu/saxml_layers.py sha256=Nghzlr9rCZ1I85-wurpziu9qItUPtV__lGeLqbnaCuQ 13319
paxml/contrib/gpu/scripts_gpu/tasks.py sha256=hGmwMmLKxfZNwORg1pzt0p84phGtmrWvRA-J6MX3wWQ 9028
paxml/contrib/gpu/scripts_gpu/tfds_lambada.py sha256=jwCx3fUsbMsDnV0EOVjMbw53UrqrlVLkf291IzGEU6I 5411
paxml/contrib/gpu/scripts_gpu/tfds_pile.py sha256=w_0J5A6k_qs8qSpKXv6UfoXyvsnkNA3qqzmVr5Bvxc4 7376
paxml/experimental/baseline_experiment.py sha256=nEgmwsr8ZURvtUE8PTsxm-9V3hWGNQ3jFBVY0kJlxEs 4968
paxml/experimental/baseline_experiment_test.py sha256=t8Qp5ImrO8wUkKfSNJOwtQn6D6P4coUHZZGCJpZEErc 5047
paxml/experimental/nested_map_config_helper.py sha256=qi1rqyxfU5RoXdYQdgteyu7r5vicr8A55SAu-CMvLNg 1271
paxml/experimental/nested_map_config_helper_test.py sha256=Bcwwpxo5KjoadHiVXw4H-Weum0S-nNqCcCGFlkhqUpY 1018
paxml/ghostnorm/base.py sha256=xdQsGpNc8hU3LsYmNbUc7imjGgIqEcm9RSrdcMJD0cc 4814
paxml/ghostnorm/embedding.py sha256=5vAxVCJ9G7VIsb6bMJAhFtrDTdY1o4EsfvzPjrIv2SM 6408
paxml/ghostnorm/generic_wrapper.py sha256=9zxNGTc71CYE2Is0nVTgFFOUFIXmoYeC-4_RJG-pE70 12969
paxml/ghostnorm/layers_test.py sha256=snuZvvv_bIGC0XN5WiuKhGmHrjvfUe4mhd7rKrAnI7U 24887
paxml/ghostnorm/linears.py sha256=1Ll3xZhC5nOSDqQr5arF1n32QJ3r_-UgtYGBXJh1Gj8 5870
paxml/tasks/lm/__init__.py sha256=Kvb2PoI3YIk0TIr1ntASQoVpYeVxpEwVj0tTxsU7uxg 597
paxml/tasks/lm/input_generator.py sha256=Kc5zsNznbukDXTf2nXJmf3OgP41S0yDhF35JdZg-MNo 17398
paxml/tasks/lm/input_generator_test.py sha256=Z_4n2vdVpNnjxD1pFlGppieD-yEYkbDyV_5pq5kEJ58 4084
paxml/tasks/lm/model_params.py sha256=yfLkbwXOVouYMsy-CKpXT69i1V3ndxCk3fIUbXvi4bI 35790
paxml/tasks/lm/params/bert.py sha256=fkeIqEVqNwBnO1apmZhweZXJzlhV0CjramZXAbfcTEo 6610
paxml/tasks/lm/params/c4.py sha256=Yqlj5eeHxrLfdFTyjYss4YFJ3HApEbWocHAzk01AN4M 30685
paxml/tasks/lm/params/c4_multislice.py sha256=qyCQpHl_2fwD_R92ATDoQxb5lvGJKUlFydHtJZFEoBs 7395
paxml/tasks/lm/params/c4_test.py sha256=lEdEP5EOPr8LIQLKPu0oJhgHdVmpbt1jX8w1wuHFXg0 3647
paxml/tasks/lm/params/lm_cloud.py sha256=zY31Q0e1iLLuWyshKeAhEaDSL0-UsopG1TE6Ar02TRk 10423
paxml/tasks/lm/params/nvidia.py sha256=F9vdf44FN9mI_frGKdrvQfXMnrId-gCk4mZQVdGBFVI 17732
paxml/tasks/lm/params/optimal_scaling.py sha256=KggpiEUYJybmWhVEHQTj1jQlQoGzlw9vlbKtuexeaeY 3040
paxml/tasks/test/synthetic.py sha256=H4oqOotmObzPseqgC7mCQd8oeQ2Exp-F8q_afsLcXbg 1124
paxml/tasks/vision/input_generator.py sha256=V8cvnug4RnKYq8qcWolO1s33ylO2XsTcfqMpLKu5jxE 5197
paxml/tasks/vision/input_generator_test.py sha256=4Tnn8YYYkI8w0zpVSxTdSUaUSLrEdzOdmhrckW1zuKo 1611
paxml/tasks/vision/resnet_preprocessing.py sha256=imHewQV5Xb6BAZxqi9ejPM_WtJO81okSLZdN7vUuuYs 7002
paxml/tasks/vision/params/imagenet_resnets.py sha256=kE39N0c4ycxxh6oTgSZvtOLqgoLaa9Oktw8rNiA45ug 9055
paxml/tasks/vision/params/mnist.py sha256=wHGlcz3S4ZP4vdbfKxsCeouVuojz2l2kb5fMFhtK9HI 8285
paxml/tools/dump_hparams.py sha256=LDM2lBQW1C3JP7MYDutt_2hshg_VuvlJGCQvSnEiCIs 1319
paxml/tools/dump_hparams_lib.py sha256=J5WAIFvxpJfLePvgf-PxM3QNDBZTOdFliYoV-unjq74 7098
paxml/tools/dump_input_specs.py sha256=5afJViiiDxI6HSl7M83HmvZbIcA59U4DwrvBbYXywl4 1590
paxml/tools/dump_input_specs_lib.py sha256=Bbv7t37ARX9o1e_Cf8eUo3JrxlcQqA3RWkdiMH1DaTQ 3755
paxml/tools/model_analysis.py sha256=1Ds6c-EeW2zjsPdeUN77Psh4RRUK_9zp_dki-jC142g 11008
paxml/tools/validate_config.py sha256=D99R5S7CKAv_sr460gCkaJGtF1cpY3yFmq-X9pjd3Rc 1000
paxml/tools/validate_config_lib.py sha256=3RfKhITls10w2sU4soSX-FPPpb1I6Rg-KQQ8w0Gxvks 4976
paxml/tools/fiddle/codegen.py sha256=Ztx_iXYx6TTYu1cENeuMoLYHY0YpDqrZIC57xv-ln3E 28469
paxml/tools/fiddle/codegen_external_init_checkpoint_fns.py sha256=6t1tBtHGe4S8KPKGLR4RNq2rXTxFJSP30AT-e1RD-Vg 5045
paxml/tools/fiddle/codegen_external_init_checkpoint_fns_test.py sha256=5tGE5VKl3ZAoyD7kJl_RvSMs_i95Lcc70T163PiZ15Q 3738
paxml/tools/fiddle/codegen_highlevel_parameterization.py sha256=hEg0F7xAZS12l6e2w2e2ItqDalObHbwCBEbI1EQqvqc 3303
paxml/tools/fiddle/codegen_highlevel_parameterization_test.py sha256=0nrhyvRmhQuGtDCWsL8riV5aear2UVjgWGWEtJA-hmI 2268
paxml/tools/fiddle/codegen_pax_code_ir.py sha256=5kkO9SZnpGVuXlueppkFQ1vHXpFrHT_hrOHdBXHQiFs 1420
paxml/tools/fiddle/codegen_sharding.py sha256=_Oz7hhPCR8weAIETt2RVcQaqw04Wwi2XSyH7Ox_7_zU 4723
paxml/tools/fiddle/codegen_test.py sha256=vZfhz07NZXVcNATw4v-SUXyC8BO0wnyC6paO5iF22bQ 31202
paxml/tools/fiddle/codegen_tracer.py sha256=6wRh-6gXt85bb0kXt7S1E34HclfRlWXfnxCiEWd_WYA 3488
paxml/tools/fiddle/codegen_tracer_test.py sha256=hPlMhhxVqwCsG4KW61QJwKBIF8euZoscdthDz9oE4Aw 5917
paxml/tools/fiddle/config_normalization.py sha256=CGJkUja1qqkc930t2WrZutdEzU_xH7q_kVtWnttfdJg 8857
paxml/tools/fiddle/config_normalization_test.py sha256=qTPl2k5vN4-Wa2G8zt0yW1rTS1lNtFX6XP2v_VJtpIg 5526
paxml/tools/fiddle/convert_seqio_task_objects.py sha256=U7JfSc61tnCw2LZ9GiR40ePa8om6eL0unZNquW2lIEM 1741
paxml/tools/fiddle/convert_seqio_task_objects_test.py sha256=v9WiEHvwlY46vI_B3zGS4ubSq5_hHZg2r3uDe0IGyXE 1344
paxml/tools/fiddle/graphviz_utils.py sha256=CjjujxOUTkaGcBQLC3Ws7InhwIQPKIWuiPUqSS0_WKM 2437
paxml/tools/fiddle/graphviz_utils_test.py sha256=TZS90mNiNspPPMS-69WXH8FYAQyDpoPtl26AqM6Bk50 987
paxml/tools/fiddle/make_parameterized_experiment.py sha256=quWUou20AMB_6e4HHPGzG6K8CiIyKSBpIEbS5FYWGoI 3043
paxml/tools/fiddle/remove_sharding.py sha256=8fBCThx8Amw2WwM2gb-IlV_VkJ-zN8q-6aURya8ld0A 2862
paxml/tools/fiddle/remove_sharding_test.py sha256=jC42d6mVs6u56XNgEBPVN28Ok0rXVXAW6tfs_WcP2mM 3709
paxml/tools/fiddle/test_fixtures.py sha256=dL4ohZmiHYZZ71qriXoMZacVknes48scWIb2Ckr0cI8 6062
paxml/tools/fiddle/unshare_sharding.py sha256=2bd7lCzE1fJY1GHPiedBsFxxN4JXCdBT2DzNn5edXOw 2486
paxml/tools/fiddle/unshare_sharding_test.py sha256=JCqBOwTJuRyyVf-5EBBRC4zt8iGxQmvJjI-nr02mCR0 4813
paxml/tools/fiddle/wrap_nested_maps.py sha256=KBv0Pnl_9mdb0NDcI66MGYReGamPe7jpQs1gnhWZTdU 1326
paxml/tools/fiddle/wrap_nested_maps_test.py sha256=4LqCdQXaacNSm61_PZXFRFe8sNtlv7ZfrImVFgN_BhU 1759
paxml-1.4.0.dist-info/LICENSE sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0 11357
paxml-1.4.0.dist-info/METADATA sha256=WMBCs1S1hJaIU2tg7eBiYhdTmO4zMb1Nr0U9P24bwro 1336
paxml-1.4.0.dist-info/WHEEL sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA 92
paxml-1.4.0.dist-info/top_level.txt sha256=hPiAZDM4XGMlFcftlHk-0rW4_hq3Nur4MaWwEiDB94c 6
paxml-1.4.0.dist-info/RECORD — —

top_level.txt

paxml