substrax/__init__.py,sha256=T-SFkjf79lhOnujiErYgA2JPmzNs-tosQjxpHn14MBA,683
substrax/examples.py,sha256=gUYIR4Orb2un-PGhypMB_2POoPgskzdx_aZXBuCI50Y,1250
substrax/nnx_typing.py,sha256=A55zjXnK0PkI1-pvpSeqepDkXEXPCiLEtWmS-FC14Jc,1071
substrax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
substrax/records.py,sha256=7PSmvOIqI4-_voW61TLXNe2RXEwakxL4dBTmc9Dro74,3266
substrax/typing.py,sha256=gObk4wi6PZAJVNHCncuKBqODxr221QlDiSidkU9Vhuw,1548
substrax/artifacts/__init__.py,sha256=pXEcp6ckZ5IPFj615skvWp4kJfcV1tRBot4RB4FqL88,464
substrax/artifacts/output_dir.py,sha256=AdeHdWnNvaEw06H-gP2GN9mtc8RX3rh359TgbJLA4rg,3852
substrax/callbacks/__init__.py,sha256=_E8NlctRLuHP60L9r1z8XiehuG27K-CrcVAjzbqM1dU,556
substrax/callbacks/_state.py,sha256=71_kJ02DcA_cvk254tQiUezvIAUsdB5jIX5ALxPSi_c,811
substrax/callbacks/base.py,sha256=qK4Q7OpYYQxorY52_KSJOWajf6H8u2WKAdnzoY2UcnE,7296
substrax/callbacks/early_stopping.py,sha256=KDYt1AyN8abWQFsN4RbhkcG3K1GSuHKX3HtWRwebOi8,5609
substrax/callbacks/plateau.py,sha256=2TRNrl71t6pI-bPPJYZR12Vod9nVFu8ia5zKwnvIoeg,4919
substrax/checkpoint/__init__.py,sha256=Zp2eX4hr8q4TAI02Hq9LUe3vR3d9x3co99Mrjw96RWs,1507
substrax/checkpoint/directory.py,sha256=J3Q4XW_uoPmwczPYusDgYJYqQuJ6TEC6Hr_J9WReU2Y,1148
substrax/checkpoint/errors.py,sha256=7pRT692y27MtAcqCw5iA3GbGfdTFpH9sZ0CWd7b7Rmo,4279
substrax/checkpoint/metadata.py,sha256=ED3y--FHX9QG6fCaR_svo3BLg3iMOczO_0q2BoiuzBI,7109
substrax/checkpoint/store.py,sha256=iYL2q5vt8Yl7IDuZhYabBAwLSoAXgZJ0t1TGnXEa2to,23458
substrax/compute/__init__.py,sha256=EuboklFSrH7h2MMUTBmy6piJGaowvzoEGLsnT8VEgBs,726
substrax/compute/backend.py,sha256=H8QJ4x6EjF5UqifowpzcP6il7fWZVyNf8KsIIiD8ffo,8276
substrax/compute/cli.py,sha256=6hGtcmEQ3G1gvgYVQyG9nX64AYtE1BQDXMScss8AXZw,6420
substrax/compute/config.py,sha256=Mf_o-W9npBdhOAClbKbRNss0jZf4sHI07e7wrqwSD9E,8016
substrax/compute/local.py,sha256=qRXg-LulF8NA12kL8aPYAKCWrrjA25F0xCLtiBUGUyc,7343
substrax/compute/modal_backend.py,sha256=Z5227550EXY30UqFdrikbd8u52cW8RnrLzKJJO_e4L0,13032
substrax/compute/registry.py,sha256=K8y3WogmM_EJaGHjUBmfv-BFBghJuqfFDuFtRGY48oE,2550
substrax/compute/skypilot_backend.py,sha256=-wVQpQF_fjyZI7WwmBePiYHKj1bGcdhk4ehSqkJUqF8,12679
substrax/compute/spec.py,sha256=1JIGb-D51vnHU3S_Zw--kQX9-pa_g2WXFAa1z__l84c,10207
substrax/compute/worker.py,sha256=q2RegXVGOGLPLy07ykn_dyU7lKsDKscNivbMOyAUH2E,12993
substrax/devices/__init__.py,sha256=BShKxDqldwX2prGNuXK4jZJEE2PpKpj_dhkTk8wAMm8,591
substrax/devices/info.py,sha256=q3oupR9HI0i3XNd6fz3Lov_-UyjsGIhZ4qhPhYiisvk,3481
substrax/devices/placement.py,sha256=7p_FLJWFfZn6pgq8Um9BbmnV841SHgmgUP3IKz7vfDQ,13679
substrax/mesh/__init__.py,sha256=H9iZ1NskvmU2xVw-fAws_2vr2r3vNuK7yph8ZnFrYJ4,902
substrax/mesh/device_mesh.py,sha256=G93xm50dvel1fOro42sqlQT_WSl6j2SuqSi5oNECxHg,6731
substrax/mesh/rules.py,sha256=-koVgPGDkXgCAAYOHqbErRomEyjJjg60V7ePXbxaTwI,3749
substrax/mesh/strategies.py,sha256=HyAx8YwqMI5UBHXV-vvXF1t1j8Mqj1ZjewiqkSx73zM,18373
substrax/optim/__init__.py,sha256=Ak6Z6uhoI6OfjJ0sRvI7FwsGqFLawaSiVh5K-992zDA,1754
substrax/optim/build.py,sha256=rGY7pBRjN2QF3HwUsDBVni2XO0WZqEt4EnwVIzIBkOc,3650
substrax/optim/config.py,sha256=3MgpVXKc9SFPym9fl9wSyrW6VhkVLdX4mrA_6ZbGLuI,5928
substrax/optim/filters.py,sha256=HIWCk3FFKSeZ6ulsJ1PrZL3B3hAP_mB6b9JZrsEsX48,3819
substrax/optim/learning_rate.py,sha256=3h0_ouH9gIvMaYBYd_ect3JdvXgKk_zP6KfQcBvBjCE,2313
substrax/optim/line_search.py,sha256=8KSpx-lsDSmdn9ri83NtzSU4105TXujov_2LHe6BvPI,8518
substrax/optim/state_types.py,sha256=F7UeSbyn-vSoPpWcqSs_2nKfwJO0u_VzvbgD-f30rkQ,2044
substrax/rng/__init__.py,sha256=DiqwvYxpmjKaRSIdabn-xkj2IXMRbUBQKAExl6YAR2I,712
substrax/rng/keys.py,sha256=Vdn22o3IwYfIEqeCv1C5un6hqoRfJtNiUZVMpWPR5A0,5745
substrax/runtime/__init__.py,sha256=WktnYYeH3jDpR_2LqBqZKq5VNLhzpxTNuMNY_wXSbrc,1427
substrax/runtime/apply.py,sha256=TcXULfYOkmUn3fPiQrbGWmnW6xryzrF6DSKpcT3EBuA,3447
substrax/runtime/config.py,sha256=vFNN_7ngs_V1NvwzJb0rQPJ6ox7roLVnrecv4BM8rKY,5041
substrax/runtime/entry_point_logging.py,sha256=NElUv4vOgmUEvG2GQrSBsdiRmkFmqrdPAs1bHKaK9-w,1210
substrax/runtime/environment.py,sha256=tlfTZPGrwRnXoX4U4cKew29u7dbf91gI-Krhup3-aNw,7894
substrax/runtime/errors.py,sha256=2yC93fYBLeBtjJW_4xBDWTsUKYGcW6EedKZTiCQeCvM,1221
substrax/runtime/managed_env.py,sha256=nBFW50Y0GOk54161o3vR42AHWrj9FWAY7sXTZVc3ne0,8224
substrax/runtime/xla_flags.py,sha256=CHkvuxgRJi1xI5IoYkxoISVxDeWsQiWWGSMi08PT1ZM,2722
substrax/spmd/__init__.py,sha256=6NyU9vKWCqaWkrpw3GN6AIO2soffRzJJ7wDollxuevc,1563
substrax/spmd/collectives.py,sha256=AzZ6FxaxGNEANmvz6hBuwugOcJLr1tro2yIFZsYwaKs,7529
substrax/spmd/data_parallel.py,sha256=JNMhZFyJNZ6MaxruRjNn5WONVTaLcrvL9yK2i7NujGc,6883
substrax/testing/__init__.py,sha256=0sKyvQgBcfwSotlGTGVGpGtUx2AUflvqGOvHM5nHTzk,2683
substrax/testing/_example_main.py,sha256=rpxtSnNXNLOWjWo5gAAu7CxEHCb72tDTh7SP0cTTMoE,3486
substrax/testing/_recorder.py,sha256=Z3xObFfyIJ97Qdj5QMoNWvHPanqpzmodk25CY9q2hKE,7427
substrax/testing/calls.py,sha256=6mwmHx9nLMHVNw975wmYLxD6vCqYZCZas1BpZDsFQ20,13199
substrax/testing/child_process.py,sha256=_i9wr15POrzYtwJ60hSCJrVHt9WqBzBLh4uLRMLH7EE,7718
substrax/testing/compiles.py,sha256=DKXHkfVi5RqJ23UHJeLxHgyqhV_3EgU7abMTPNi0atY,3414
substrax/testing/compute.py,sha256=ZhmUwIcjDjsA-ZNwCEy_OeQvgAM8NGcrVfYtdTNPhCw,8792
substrax/testing/examples.py,sha256=7lj6TKk-Yxb7e14WYHQUMMt9MiowOLq78ilSS4SdOoI,5568
substrax/testing/gradients.py,sha256=Ft1IK83YkfnBzNBrgENqV-00J0WkCAxyivS0kl29d-Y,4824
substrax/testing/jax_calls.py,sha256=7AS1VQQ4ITM8e4wzGnwdfDqjdikyqO9hSNIfzrKM2_o,2856
substrax/testing/jax_config.py,sha256=ZAvkpBBsqXHHqwX7UhOjGJagWeqdJsTTkrypb7CBPLU,1627
substrax/testing/pytest_plugin.py,sha256=c5pR51rJUA7S4Die42TofXkCW6W3uxgAHsLBs21rZ00,10511
substrax/testing/source_scans.py,sha256=vBoDOhk5V6lNfj11Bus5s_UaG91d7MdrubBPr-15vag,9017
substrax/testing/traces.py,sha256=fusYnxGcnul9qdJQMYMcckmXk_SDzZIjeB3MEbiFUQc,3347
substrax/tracking/__init__.py,sha256=f5xyj9WCJ2v2ZGNJQxJCPucOnckSySMfVXO9Os7gS3c,581
substrax/tracking/_optional.py,sha256=cRXwlHjR1FE_J9VU5Y620W8RKYRLkks3K2cl7HPUrmQ,824
substrax/tracking/_plots.py,sha256=0wJaTA-PBqWD-_oKoMl-wgN271I9m7Q4HVxaX97BMPI,2372
substrax/tracking/logger.py,sha256=3E2cVyZ8U0--GOsSA8m1Z2_05yWRNUFPLlWwXnR6_vo,13759
substrax/tracking/mlflow.py,sha256=FOZgsVdbc1CoGe5oQHteot4Jbch0AFV-qIJuf8Y0upA,8965
substrax/tracking/wandb.py,sha256=Z3DcLqEL6ZT27VXfRAH2LautXHjMB1P0Ugm2h_P2_fI,6428
substrax-0.1.22.dist-info/METADATA,sha256=1WT9KEdfQyPE3rZl1yBIbfMtC6NJ7vWe3YCz37HHQVY,32021
substrax-0.1.22.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
substrax-0.1.22.dist-info/entry_points.txt,sha256=Lkn6LpCpVAKB-DDiz_5-eJfOf7Wzyj7NwCBEUhTjzNY,249
substrax-0.1.22.dist-info/licenses/LICENSE,sha256=wm1uGOlpkgBGzAsTT0-iZkFq-k5O9Ty-3ZITpUwpBLc,1070
substrax-0.1.22.dist-info/RECORD,,
