xtrax/__init__.py,sha256=P3JA1gPmDf-qjqrsO_E-tu3CD33yiMtvyBpQX8donms,3110
xtrax/checkpoint/__init__.py,sha256=KOtaTnR-yCdfTqMD6hNHpcqhPzDUehtr0ZyjPMF0GD8,239
xtrax/checkpoint/orbax.py,sha256=5d8_YTPEq8KneAimSabpweaydAdTV7N6DHbvlQzw1Lw,2898
xtrax/cli/__init__.py,sha256=Fnye_PgP2J4AkweYoRcc7Ou60Sv5dLPJifM23-cyAtU,914
xtrax/cli/config.py,sha256=UlEKNVFBgMFwR9WwsiCIqbLIzHSIXZMtA5WTScg_jEw,1656
xtrax/cli/emit.py,sha256=bUfqlvc0g431GLMYbFpvmQJEwCytcKpzDjYTH9aKp0I,5485
xtrax/cli/entrypoint.py,sha256=1xUZOv_3tFfChwKhEcmh2T5kU-xcbbwvxc739_iPkWI,2328
xtrax/cli/errors.py,sha256=PLh4cN3hUdr2rf9k_oBpSJzpfzj-OqJH1IiIwgM3afc,1477
xtrax/cli/explain.py,sha256=h8Z9chFulrODd3CaVBu9NAmRpUKrjqW9Rkv438Bow6E,4537
xtrax/cli/export.py,sha256=VRNB1teO5cQZtC88M9z_G5klYOe3niVR_TN2vxNnD0Y,5966
xtrax/cli/hash.py,sha256=_nxA5pUISand6tjdFYdwY6BA1VkjZO000Iqecl5EKL4,645
xtrax/cli/loader.py,sha256=U79y3Hh3blHfQCVIx0FGeKVQveHpx1h_jm3_DG97hjk,2463
xtrax/cli/manifest.py,sha256=3Xf-FNq6la5fBAYzi7RXFLvuFhBHJHrVULcSc-mrjzA,3999
xtrax/cli/plan.py,sha256=ZjhmcBB_-mdSWXNzZQcaCRPqnQkU-pvGC_XpOtIch8o,3909
xtrax/cli/registry.py,sha256=SyDa9zm_l4DahHyiReT03AKUqTscnD0HvuLj1EgMpIM,1087
xtrax/cli/resolve.py,sha256=-_Wr3P8TRDzJxLTdQKMAXp4GqUv6xF8_JBEtVpvpBKA,1542
xtrax/cli/resume_verb.py,sha256=-C137-RdchlmkuU98SP1ltgzFs_tgw0XIjjcnqVfI-c,3008
xtrax/cli/run.py,sha256=XEBXf31CVuki9Yvez3RqVS17iGb1XHOd1E3Lz4H85YA,2900
xtrax/cli/run_verb.py,sha256=VReGfEEWCoSYgnMXGFjQcJqp2QbxwoWQenHfBlFhWtU,511
xtrax/cli/shapes.py,sha256=hVlBYByBFHKn5kHuoJ7hZn60Tnf8XV7FDNavSfeyolE,6595
xtrax/cli/sweep_verb.py,sha256=dhrbNaTqf_HQ8cK0oq4rIoNXGn-GwUw2FzNGFGDKGhU,5349
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=PjgXgculeY2pYrhLz8-J3l24kmSk5JOwzvZV_pmyydk,5934
xtrax/distributed/sharding.py,sha256=zQ_x3Gmk9ZsLQZtMLr2HGbdboN_7_TWJo04dE_0Qk9s,5967
xtrax/eda/__init__.py,sha256=RH-DX-Mgwg3cNmgVa4YZpfOXDlNQQ8Q1Lap3i5vctIY,1423
xtrax/eda/explain.py,sha256=rioSrCDH4GaPVsuTVPmdFroJYRwq-phQE9GXsixRNWI,1468
xtrax/eda/export.py,sha256=TTNARCsFKWbbW-Zs87XUByYGW7tqiZKb7Kg22orMiQI,4938
xtrax/eda/stats.py,sha256=gfLLiIqYTUkjXdzw6eMnrktRiYORZYyi2Ow6XUv0RKw,6473
xtrax/eda/types.py,sha256=XcuTaiKYSvkaE6-qVRSsICy7J4VMP-b7c1cQHHGKlq4,7315
xtrax/eda/viz.py,sha256=XkPvB8JZAybaSf5pqz2La0CnsjhHKKYe2YYTD1GQ074,14364
xtrax/engine/__init__.py,sha256=t4mYlxEbXW0MjY6FjhsTAaptxhRTOLAh0vbWcwp2Hf0,201
xtrax/engine/engine.py,sha256=OhY8ZIXnjSH_Pwj0uN1r6XSMvpig04rodSMKZC8D4ko,9269
xtrax/engine/io.py,sha256=O5AQf6tJULj9ZpwpY2TuLWytVfbSyEXiehwRax3n4sI,6323
xtrax/inference/__init__.py,sha256=pMhtgtA5QeZXRLXwC3fKkuh7B0t0bMw5fuIe4t2IzMs,696
xtrax/inference/abstract.py,sha256=WHsH3hYS2M0U0rxxOWfYqSiUieYs4r5MRQvYBEPp9Cc,3883
xtrax/inference/api.py,sha256=06pXnlG6qiNqMPAo4ws1jjtvwKNi1b7ZiMHG5PJ1qGo,3485
xtrax/inference/axes.py,sha256=PhXxfxWpVRFhNdREPq3Xw794d-vy-_jUdU1NDXaN1CM,4290
xtrax/inference/config.py,sha256=ZNLUZGmFX25YQ8fw18UwSmhRH5THJXkTTYM_UJImWII,3022
xtrax/inference/errors.py,sha256=MhPoVdg_ZBOKjtMKFH3eup5xn9NcCX79x7IllGWiRj4,1319
xtrax/inference/schema.py,sha256=oAaIsPk_ViZqkXUmEjY9YmPWCGCmw3aFV7mLj7xh35U,4858
xtrax/inference/verify.py,sha256=N5h2f3Q0a6yPdkwUbiUgGGSr0BwLKuRb8as18Im54Xg,3634
xtrax/io/__init__.py,sha256=TPp2sJivgX8wshx72KyM6R3pt5nbnrTrBe-Uva3fapM,180
xtrax/io/callbacks.py,sha256=_mLKaWg3jyKiprT6191PAQrLpMm4PoAp3LQCp_cHmxU,1097
xtrax/run/__init__.py,sha256=JvMOZUF_XPV1FQCG_YNKwRyQ8CF4INVFG9ovVH0DBBw,287
xtrax/run/resolver.py,sha256=g9TL02FtJIscgR7coMSH3Vj908KLYGvw_-_7A-ycC8M,1206
xtrax/run/sink.py,sha256=HQ9yC6-QXUV96bfdqoC-sxJMSOkFyMAhqwfMu_pxbB4,350
xtrax/run/spec.py,sha256=xGALrpP1j4A4RMs2xudHWghVIMl7K--JDtOrenlkdOo,902
xtrax/safety/__init__.py,sha256=vy_PZ2V_Z_k6hB0ZHbysPU6hBCABQJfD9umccsrO9EE,100
xtrax/safety/manager.py,sha256=xHAd_7MnBna7aCsNRm0OXsFWfxLjEU51BMW3-gBpZTo,2079
xtrax/safety/ops.py,sha256=yjqQb5zMs4oFH0odg9HHoL15vGWEYVt0XpQYmbGCj-k,1379
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=zxi6jDAHKkIWWfOnxsrFLu4Um0fdS2iD61NhB5GJRr0,6819
xtrax/sparse/manager.py,sha256=oRowbQS5LdPLlY78Tz6aBRjc9mPYdV2XGBpTHOF9C9s,1788
xtrax/sparse/policy.py,sha256=7M2IUR-Hg57I9JoTWMdJpR73fmuLbTeJMDODq-DQThk,1933
xtrax/stages/__init__.py,sha256=AuGdRlsCsnW-g_sj9Gvh2xeIcsUBnUKFy-GHTwfVfN0,803
xtrax/stages/boundaries.py,sha256=L9acitwmWFjQ2KjpchRL4fHxdMscOaz-pVKgyNZKdZA,3489
xtrax/stages/bundle.py,sha256=AtB5gy9Q8D1a8XgE-SKKTYVRXLi6tt2F-mQVWpcpN3s,4559
xtrax/stages/protocols.py,sha256=2IN5RYVgv6kewIuuNuPXcXTeX1Lr5a3PpM8odl2GUX8,1400
xtrax/stages/topology.py,sha256=f6DeGzeGFhsLfB026ykIeWraSZbu0z7RHKG6DVJnL9g,4715
xtrax/tiling/__init__.py,sha256=248LPex9MMyvAXzkS4XrYNCsEtGPziN21oo2D3mPlyA,1674
xtrax/tiling/_plan_wrapper.py,sha256=-AyHI9So80WDSJmY1_5d7oS_x2Tcdii-YMaxwsHXBY4,831
xtrax/tiling/bucket.py,sha256=uPD_uIrf8FgsZj_ya3XA98K1Bx5DdyEgzoCm3Ngb6rY,4435
xtrax/tiling/carry.py,sha256=2Vd3Zez9BfWtOenHqpode_ceaypN-PN6jWLK3OE55JA,1673
xtrax/tiling/carry_shape.py,sha256=tS7mbyfvb_lJ34PnTw8mye_p-L-yZxEbyOgK7cSG1i0,1532
xtrax/tiling/dedup.py,sha256=cMWNRvzKMXNEggzqV3R7kJdZB6anjeSuCQEXJSAPjt8,5393
xtrax/tiling/dispatch.py,sha256=onWzYNlgAbySHA2gXNsIAaIOaW2a51Th61okKxD8Qoo,6785
xtrax/tiling/iterator.py,sha256=6dlPE2Elt1GcQqXs6LRU-VEWNzXhXfh5fHzqxqZZUt8,8056
xtrax/tiling/plan.py,sha256=mpEO3gyjuOWsZDDhY40rgIQf7SkwZWz_EIzBAWmbEOs,14987
xtrax/tiling/roles.py,sha256=CY4PBts8wVcVrJftdvD26L2OAoU6XTIVBhnpZ9gC8cI,1616
xtrax/tiling/strategy.py,sha256=3oGS4m1JGYKl8UeAw69A8OlNFDPlrD-wgdZgmDnO2II,3523
xtrax/training/__init__.py,sha256=yv_IDjst97P04A3gmSAJ7gM_hE6xmndA40ZZjQzo7Co,656
xtrax/training/grad.py,sha256=ACmh5DttMAut-pQXqMiSF_QmsMMuYG3sALdyOSmOe34,3702
xtrax/training/loss.py,sha256=Vgf1dNRaZsF5UdSMw4DLSJW_AG6l5tbCcE9XuoGg90A,3775
xtrax/training/optim.py,sha256=-OHAQAsrswiuyjOASfVbEFJkRYtVyruRQ6PeKNIpXiQ,4266
xtrax/training/state.py,sha256=bMhcPeJFc6TnH-F51AxCFn5pLADm67I9CUExzKtlo2I,459
xtrax/training/step.py,sha256=NUnOCHGEYYe9IkrdbPQKKWwdNAOioBEc-0p3a62Umfc,4707
xtrax/training/trainer.py,sha256=z2xQCWJnfKXA9ClP_hJ-g2Y4HWGqsEFZg-B8fq1JKgk,2598
xtrax/training/types.py,sha256=_uwE-ts37GUojxUlconDNke-Dl2F72AV_5RfmTptlpI,1694
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.3.1.dist-info/METADATA,sha256=iZd_y3lAARZfWBKVMhR0gUZy-X0LIELvptb-eV9dvjA,6796
xtrax-0.3.1.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
xtrax-0.3.1.dist-info/entry_points.txt,sha256=OjgaCHb3ru4Pk34KBAdrO4JvEKj_1DXznJaMOesR0WE,41
xtrax-0.3.1.dist-info/licenses/LICENSE,sha256=hb-Z19QpgHhaB0t8PtNaGBOHHnrFIF47yEQvh76deZA,10318
xtrax-0.3.1.dist-info/RECORD,,
