xtrax/__init__.py,sha256=VHtabjusAwvNLCzsfVIZqWtYY0RT5zImWRz5v3hM_Rw,3273
xtrax/checkpoint/__init__.py,sha256=KOtaTnR-yCdfTqMD6hNHpcqhPzDUehtr0ZyjPMF0GD8,239
xtrax/checkpoint/orbax.py,sha256=5d8_YTPEq8KneAimSabpweaydAdTV7N6DHbvlQzw1Lw,2898
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=eMinqUwNVY3ZUXZ1hwnITFuf9vLiPoYqpsU2RMPdhlI,1261
xtrax/tiling/bucket.py,sha256=TexrNSH9iVZppdtC3Dp8I3KHgNWhyMtTEi7NHiCD8zs,4457
xtrax/tiling/carry.py,sha256=qX3qsori_pwtG2Q-2JOkvreqJJDQS6eawxgz8RpXetc,2153
xtrax/tiling/carry_shape.py,sha256=b5jW18k5wbSheDO37bK8ZHFL19YRWgrWxeLRMW7yAQo,1450
xtrax/tiling/dedup.py,sha256=uW5Br4AZ0XbMa821gabf3qVSJvP26qiry2QpxnMqS-E,4171
xtrax/tiling/dispatch.py,sha256=U3cRfyEr7cHPkZX5FRL2gIxN04btqj1gE3lg-xHEs2A,6000
xtrax/tiling/iterator.py,sha256=HmHcn1cEQM8uyXZr4OXbLzu1y2kmKmdIGpPIKXWiEb8,7416
xtrax/tiling/plan.py,sha256=_A9cVqtQaOy7erUqiK-yiRtLtzZdpnZqs333lSVMY2s,12808
xtrax/tiling/strategy.py,sha256=DmBi-IKGZeWuDdCC5kZyKEshrqLjKhMEMNbaq22LpV4,3520
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.1.dist-info/METADATA,sha256=CGcrvaJ0bdsws0jvOHr-pl0TaNbXQys99N2_lrZ59dY,4431
xtrax-0.2.1.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
xtrax-0.2.1.dist-info/licenses/LICENSE,sha256=hb-Z19QpgHhaB0t8PtNaGBOHHnrFIF47yEQvh76deZA,10318
xtrax-0.2.1.dist-info/RECORD,,
