xtrax/__init__.py,sha256=oVVI6sGpUJkfJuz2Q6AMU0yFBIS1EnNBCowiU3GEbAg,3121
xtrax/checkpoint/__init__.py,sha256=KOtaTnR-yCdfTqMD6hNHpcqhPzDUehtr0ZyjPMF0GD8,239
xtrax/checkpoint/orbax.py,sha256=zcfveWvk0KhMJ1mc9i4_AJDTiJB8J2B81y3J4nDQFHU,2875
xtrax/data/__init__.py,sha256=db_Yfkol4HQlT6v7eGdBTClQBhu9cxE8iJDxb1vxFDs,227
xtrax/data/module.py,sha256=YS4qDcqS9kFj_R-FTKi_7D3EHH3qlTNYCcWOTgn7ICg,1266
xtrax/data/pipeline.py,sha256=l901aSHetjMq-Jz8HW8sH5FTJwvp0EbNxY-yTzyG35c,627
xtrax/distributed/__init__.py,sha256=7fnuYwj_PuAWdJLwHZFjOxFwrKe24wf0OKwW2HvLGyI,355
xtrax/distributed/init.py,sha256=av47lpDuLv7ocTG9ab-YngV8Bv4iO882-kUhfabjzPk,5980
xtrax/distributed/sharding.py,sha256=zQ_x3Gmk9ZsLQZtMLr2HGbdboN_7_TWJo04dE_0Qk9s,5967
xtrax/engine/__init__.py,sha256=t4mYlxEbXW0MjY6FjhsTAaptxhRTOLAh0vbWcwp2Hf0,201
xtrax/engine/engine.py,sha256=W6x3JEzH-lOlovoXUP9M1iXxlFDCVy2Q7Kfp1ox7ym8,8360
xtrax/engine/io.py,sha256=O5AQf6tJULj9ZpwpY2TuLWytVfbSyEXiehwRax3n4sI,6323
xtrax/io/__init__.py,sha256=TPp2sJivgX8wshx72KyM6R3pt5nbnrTrBe-Uva3fapM,180
xtrax/io/callbacks.py,sha256=9stV3uZcy7j3eibv2zI8cwSIDan9HIsAt0sxbFxl5n4,820
xtrax/safety/__init__.py,sha256=vy_PZ2V_Z_k6hB0ZHbysPU6hBCABQJfD9umccsrO9EE,100
xtrax/safety/manager.py,sha256=xHAd_7MnBna7aCsNRm0OXsFWfxLjEU51BMW3-gBpZTo,2079
xtrax/safety/ops.py,sha256=ZivYpZKtrpBwBj2bL1t7cFqjFRFFIY13f9b8JolQvWk,1238
xtrax/safety/preemption.py,sha256=9nmERUdNjWKdgoB5VS9IJzAzSD315sR7oIWt9Q5bOc4,1739
xtrax/sparse/__init__.py,sha256=ooi-dF95rIXgUl5QwIF678N_p_4hphgu19idX6bB5VI,408
xtrax/sparse/config.py,sha256=xvwNM87QXf9qJEQ_KTMNfHUkryLJd5pX2hbh45tzBSE,458
xtrax/sparse/inference.py,sha256=O3tWfyf9N2Bo0KvRe67oYhNU3-VMx2E3HLqXCU4H4Kk,6874
xtrax/sparse/manager.py,sha256=oRowbQS5LdPLlY78Tz6aBRjc9mPYdV2XGBpTHOF9C9s,1788
xtrax/sparse/policy.py,sha256=fH5z0GtOvczzX7334oZ4CpQ8qNsj_fnTneSNUFENCMY,1985
xtrax/stages/__init__.py,sha256=zkDd4WjB22D4Wm0gV3XyknzaCF_zrmeWih3DowvMbLQ,163
xtrax/stages/bundle.py,sha256=X3-r9UE7iDeWV_emIz770Lug-Rn6QOXtm23RcZZtItI,4643
xtrax/stages/protocols.py,sha256=zEVT7PPelOxGDesMSsuWOHtjZ19etISsw9bKusPj8hA,1199
xtrax/tiling/__init__.py,sha256=Gu83Kh69h4Z3ggyZ7C8LlkrlO2yvA-hMb2r3xtAkV1g,544
xtrax/tiling/dedup.py,sha256=95Qs5nHJ-ryzCtrAR9UcM44zfxRp6gReIgdxPwTSyJE,1688
xtrax/tiling/dispatch.py,sha256=9AcP67SBP1lyPU3j9lTiawmnD4Yz8GoJ8Tizm84u_SI,2397
xtrax/tiling/iterator.py,sha256=q86zz_vqdVMHrJvUECeawaTr5OyqDm-29AAJITsi9HM,5139
xtrax/tiling/plan.py,sha256=8lULc0JVK7nuHksrk_E5mM8DgOSgi_oThn52Osaojqo,8150
xtrax/tiling/strategy.py,sha256=DH1IpCmFfsys3xYhMo1vn4CKC6yRxo-JfX5H1uRztVI,1447
xtrax/training/__init__.py,sha256=QtBBMJfS61t14zZcufrFC4KUm6SdMhiolq9ZNX4-ZRc,594
xtrax/training/grad.py,sha256=ACmh5DttMAut-pQXqMiSF_QmsMMuYG3sALdyOSmOe34,3702
xtrax/training/loss.py,sha256=m37mz5cRCFpKiOWaPUDW-aIpPjOsoTzDSBL0sf3np0E,3789
xtrax/training/optim.py,sha256=9HMJh8RRvCxmz-1YKAvIIkHUtGUq5LkI2qcsD62GPnk,4292
xtrax/training/step.py,sha256=rMZWHvIAwrUfAYY0A5WEnsQvyMfqGuYO1KqydRegavk,4729
xtrax/training/trainer.py,sha256=kWy3RAlYBKSMoJq9p7zH3F8RwIgKhfF6EtCkwav_JwE,2620
xtrax/training/types.py,sha256=fhY3aJij8YWfhehNnedCJYUSUMrjtsyCxIJYjxAKgCs,1708
xtrax/transforms/__init__.py,sha256=Z0zVwreJ1jwaV6huRbXrvDk3W6TfwtweB4EnWCfxu0o,123
xtrax/transforms/map.py,sha256=X5vPZtHYDWVx0M5lGNxOCqUNz-4PKW7-jZdet2s3kfY,1302
xtrax/transforms/scan.py,sha256=GmzOj8I7eqG_NmSyRoN9Vk3xEaxvB0hLzCmqUWpz6x8,1695
xtrax/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
xtrax-0.2.0.dist-info/METADATA,sha256=jg_-ndKENa6cgigTLWFknuG-Yp_3QW8YwJFIPVSQMzw,4431
xtrax-0.2.0.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
xtrax-0.2.0.dist-info/licenses/LICENSE,sha256=hb-Z19QpgHhaB0t8PtNaGBOHHnrFIF47yEQvh76deZA,10318
xtrax-0.2.0.dist-info/RECORD,,
