substrax/__init__.py,sha256=QSN5kGaUItG1vPErA72iImf6fv6XWUplnTmvP55TEm8,598
substrax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
substrax/typing.py,sha256=qSx3sXG1FEyF7dEdreva0Xm1tn9nyXoF0mlzZBz4ATk,432
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=admYkt31u17yMLPL_U9YlK7vplJBVcAURnWWv6Sd_74,1961
substrax/checkpoint/__main__.py,sha256=lwG6TsDlUdiVD1CXgGxbO9ofDqPOKUc0slvo9x8BP5M,1729
substrax/checkpoint/directory.py,sha256=J3Q4XW_uoPmwczPYusDgYJYqQuJ6TEC6Hr_J9WReU2Y,1148
substrax/checkpoint/errors.py,sha256=7pRT692y27MtAcqCw5iA3GbGfdTFpH9sZ0CWd7b7Rmo,4279
substrax/checkpoint/legacy.py,sha256=9aB1ytWZf3-Rlp4azA_OtM7Bntx7xWwJNTnbL9HfBl4,1525
substrax/checkpoint/metadata.py,sha256=Tgdj2TBPJvZFEQb2hgsc48iSxJOCvNaMD5dcH0bMHjE,7546
substrax/checkpoint/migration.py,sha256=JRyk_yrgOYTRS1rVfumcc8HePu-xsG0oAXYiCYiZuEs,5926
substrax/checkpoint/store.py,sha256=EIOi6phlSDMPwTXw-acmwCeZT-vrPgudEPpaWtxWXb0,26079
substrax/checkpoint/upgrade.py,sha256=gVDpJCJVgFjHCKm91IKKZuLFJnc6OYRSrscV_lqbfg0,2577
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/managed_env.py,sha256=nBFW50Y0GOk54161o3vR42AHWrj9FWAY7sXTZVc3ne0,8224
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=JvXzrHfzP6h8-WTVGG-Bzz1AWeE14ZxX82cdAYXAUys,1819
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/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=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.13.dist-info/METADATA,sha256=Pbej9Sc8bjwKc1hG5mN-6l8Fswoh0FsjbuJeGyeETPQ,27098
substrax-0.1.13.dist-info/WHEEL,sha256=THafob7ofN-NsuMN7Mg4qZyHaQI7KkD-QlcQatYhXPo,87
substrax-0.1.13.dist-info/licenses/LICENSE,sha256=wm1uGOlpkgBGzAsTT0-iZkFq-k5O9Ty-3ZITpUwpBLc,1070
substrax-0.1.13.dist-info/RECORD,,
