substrax/__init__.py,sha256=QBRBRYvUXjKABQy4QHIy1d9U0agqRWaWd7lF3Gc22f4,598
substrax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
substrax/typing.py,sha256=Dc4DUr-DnHjMRZxTag820B7VShyDgnHlA5s0WT2gXyc,300
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/base.py,sha256=OTem2oYlLSNQwxqqmGfk5wOZ148SL8FKQgDnCAXi4Jo,5708
substrax/callbacks/early_stopping.py,sha256=qzUF8VvUZtbqqNB68Z6UwTkr7Q2QeaH5OPf2cO_XxMY,4362
substrax/callbacks/plateau.py,sha256=N3mIP5a3CmxI28jtDoChR3C3GhANI3aXKWbtvussm4Y,4126
substrax/checkpoint/__init__.py,sha256=yXrTKSz8ub0fgykeK4d2zEIKBI6zLBH4YE_w1Z26d9U,1851
substrax/checkpoint/__main__.py,sha256=lwG6TsDlUdiVD1CXgGxbO9ofDqPOKUc0slvo9x8BP5M,1729
substrax/checkpoint/directory.py,sha256=J3Q4XW_uoPmwczPYusDgYJYqQuJ6TEC6Hr_J9WReU2Y,1148
substrax/checkpoint/errors.py,sha256=ByLBGESf6rEIdoCqRt_kRyGgpfZKWimxO9ucrwwoQ-Q,1852
substrax/checkpoint/legacy.py,sha256=9aB1ytWZf3-Rlp4azA_OtM7Bntx7xWwJNTnbL9HfBl4,1525
substrax/checkpoint/metadata.py,sha256=5pCv5BNCUsW5fzYagjNRxGtd7Svc1i--60soZ7lKWzQ,7599
substrax/checkpoint/migration.py,sha256=JRyk_yrgOYTRS1rVfumcc8HePu-xsG0oAXYiCYiZuEs,5926
substrax/checkpoint/store.py,sha256=IynUbTsrucNDDIpsRoXUM3ZkACV8Rmae07l2gFfCGcE,17899
substrax/checkpoint/upgrade.py,sha256=AdFsEpL-apHwcbLpQt-4u7r-KgxYN5O7YDNE7EFcdLQ,2138
substrax/devices/__init__.py,sha256=dohqpkfBaLsHuTHX1cDCMWrGJpCorRJPidPKU3_luIw,496
substrax/devices/info.py,sha256=D1sD5IREz6AnZpp9_bImux1wgzG9bl7V1NZtTNFdo_Q,2801
substrax/devices/placement.py,sha256=rQSyJob7xaarqp-3oQmTCHmiplciuANsVAMn_FtujOc,14524
substrax/mesh/__init__.py,sha256=H9iZ1NskvmU2xVw-fAws_2vr2r3vNuK7yph8ZnFrYJ4,902
substrax/mesh/device_mesh.py,sha256=oZgt7G0fwz9Bdutbmg4BX78rLLyyZhtQtgEIeJHLVA8,6648
substrax/mesh/rules.py,sha256=-koVgPGDkXgCAAYOHqbErRomEyjJjg60V7ePXbxaTwI,3749
substrax/mesh/strategies.py,sha256=0kfnKBWzIUOtHNCy_5O-9IOG8T4C5m957Taav7aQ_BI,18661
substrax/optim/__init__.py,sha256=nXjVccspCCuiXJj6undO37ry-VeTM66vDkdLEFLrsC0,1226
substrax/optim/build.py,sha256=raHmwv94ZKLP4dLBga81XGfEILk5EEpf5y1c6rgkRIg,3670
substrax/optim/config.py,sha256=L6EuB1E7jyeBsIsTJwbuKm9Fu9554zhw2iLtZUdh5Qg,5967
substrax/optim/filters.py,sha256=wokaVPK6m-G921emAU9hgtIIKTnmrrwT6EnvMVb9H3o,3657
substrax/optim/learning_rate.py,sha256=38Wdy3tEIYn_ec8Mm0O234Oeczx8S27qrJKjF99XnV8,1913
substrax/rng/__init__.py,sha256=DiqwvYxpmjKaRSIdabn-xkj2IXMRbUBQKAExl6YAR2I,712
substrax/rng/keys.py,sha256=Vdn22o3IwYfIEqeCv1C5un6hqoRfJtNiUZVMpWPR5A0,5745
substrax/runtime/__init__.py,sha256=r-fYKcHbcqr58eo7b-FE4SFYLlcZgZowiDoVx3C8sVk,1298
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=GY3BlQ_NoC-RpP9Xr5ut5BjpH6lWantYnrw-Jo1Neb8,6364
substrax/runtime/errors.py,sha256=2yC93fYBLeBtjJW_4xBDWTsUKYGcW6EedKZTiCQeCvM,1221
substrax/runtime/xla_flags.py,sha256=CHkvuxgRJi1xI5IoYkxoISVxDeWsQiWWGSMi08PT1ZM,2722
substrax/spmd/__init__.py,sha256=GiJCbKOovvu2aWTBXbqePnFURGNgFTviDKCaEyuf3GQ,856
substrax/spmd/collectives.py,sha256=NFQsmrdGTVuosRMLcpWvoH6fv85r8KYPjAEJKKTy_-U,6670
substrax/spmd/data_parallel.py,sha256=3ap1k6BuAK4RSvZ41anF2m1KL17yvWPMhMzV4WUhJB8,5406
substrax/testing/__init__.py,sha256=W3gkvr7ui_eR9T1tE-jYGcN2KNLjZBmsYjNLqzku5HM,1634
substrax/testing/_example_main.py,sha256=rpxtSnNXNLOWjWo5gAAu7CxEHCb72tDTh7SP0cTTMoE,3486
substrax/testing/child_process.py,sha256=J-iAQeU__V1yiUBjmuI4RK5Z4rjnr7HLbZentqFG75s,6701
substrax/testing/examples.py,sha256=2LM3Yu12wHuLySivVOZw9W0u1MSjgpZ83CVv_3xdgWY,6426
substrax/testing/jax_config.py,sha256=ZAvkpBBsqXHHqwX7UhOjGJagWeqdJsTTkrypb7CBPLU,1627
substrax/testing/pytest_plugin.py,sha256=A6sSD305WTyAhbWcXZvLzkYs-Qpvd4yp6faB5uu2EUA,9133
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=CslvVzrrqY5K32vkIgLOZCBXsvwR-MA8rnl8w1VGbHE,2401
substrax/tracking/logger.py,sha256=c0tMrTvrOwb5ywt3d6216rBvhp3uacGjfNcX5z-32Hs,13855
substrax/tracking/mlflow.py,sha256=r4VZwwf_kpNnfkmKNbu5Xc978Nreemc2BwN_dheU6ho,8796
substrax/tracking/wandb.py,sha256=LzGuN6tA67g0vHXdwgF0eYGu9c7Vwo9zcWtWqskKcFI,6269
substrax-0.1.10.dist-info/METADATA,sha256=lBg1GhJiM0AP7ObovZ6bvacdgwU5cPiSL-Rt_MRAASs,27058
substrax-0.1.10.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
substrax-0.1.10.dist-info/licenses/LICENSE,sha256=wm1uGOlpkgBGzAsTT0-iZkFq-k5O9Ty-3ZITpUwpBLc,1070
substrax-0.1.10.dist-info/RECORD,,
