torchax/CONTRIBUTING.md,sha256=gbB2ewxDLC-HHRBC3B8HdppV_d9MbDd-9rvzGQt7vZU,1440
torchax/__init__.py,sha256=2trSDOtbBY0V3A-lwhuOF0o3fy8N_mL1SIT4xrUTnzw,4159
torchax/amp.py,sha256=h5sYm7jhbQhn8kmtdd8ahzrmH06r2VKGYigHQzRzmZc,12366
torchax/checkpoint.py,sha256=3nDcGOVbeeBzoaI1dfQxpUPdNiRuGGkVj65DlXkEpF4,2437
torchax/config.py,sha256=c2JVtKx-GkkQn9vGxYgweGrm57G60mH1BB-SfE6-d6Q,1497
torchax/decompositions.py,sha256=yR2QXWbwFS3Gb9Wbc318fJDX04wLYc_TCmclafOmqjw,29434
torchax/device_module.py,sha256=7WrLUBjMQiAilVfRwEwJrbfkizPZAC3022UO70U5uEQ,924
torchax/export.py,sha256=ztEECMqORuKX75VvQfRs4qwso1NEhFwylM9j6hVShJM,9544
torchax/flax.py,sha256=LxXS0ZKoEU_hNmweyuNSeErX9mThbkI4El1nAUBuGic,1855
torchax/interop.py,sha256=MS5BGmXhiucKA47EMOwJxpHhuwsQu19EyZsGz4zSw9g,11845
torchax/mesh_util.py,sha256=9QPU7fWxVnb8v2A4S1Accu6PCAeEsNgaXVz1fIfhXeI,9889
torchax/tensor.py,sha256=lEhIPhpJutnj8d7urFfAZdjihdh4iJdozI0bkJBIU-E,21900
torchax/train.py,sha256=3Uc5zMxHAp-qjwdIm-ebjsYrEC7KF2y-2JKFsNkHhPY,4726
torchax/types.py,sha256=IiHQHVNWQvo_6sT2Q3j6YBbc5p0DMrewrXNQB0fZb84,963
torchax/util.py,sha256=sLGXkmYE5dOBC-3Co2HxPTsjod36l9bPvzhufEw9kEE,3644
torchax/view.py,sha256=PF9ti7eLKBb5-kfx96lijlgSlSBXPCPPtGt_12iMiQI,11696
torchax/ops/__init__.py,sha256=VY8LjEJnLhMdSbPaTOzrWFn_nFRseWOSEqLFijyX_6Y,845
torchax/ops/jaten.py,sha256=NvKxA_Cjp0GF7IQidHrpMkOG1kd0vFW11wfe4ZdUWiQ,177190
torchax/ops/jax_reimplement.py,sha256=kjU0Fopq_zM-8GJDSE4nPASAG-8es0I108ran45d0tM,7877
torchax/ops/jc10d.py,sha256=l7-bi-uFLbTDjHKsE6H-7eO_PG3j0j1S1vIFEMAyvII,1897
torchax/ops/jimage.py,sha256=6YpFBaYSvsXuYgsZY2Y-c5yi8bAROxlRCcXSFQMau0M,3757
torchax/ops/jlibrary.py,sha256=ol12qM2qP1qfqXnmWcVCpPOHmLODZkhiFZmxZGJAD6o,3505
torchax/ops/jtorch.py,sha256=SiXz_OQyQ7XW8uJ1i9Fx1n-G6k2Ws23fbrFsRr3fzUk,19191
torchax/ops/jtorchvision_nms.py,sha256=OZfYntyIatgXtMBU2hywlWvchMKaK028nNsFE2fCy7E,9478
torchax/ops/mappings.py,sha256=RP75HTKJYfZ08dTByfUnCEJRP42jDHcoQXQrBafWI8M,4362
torchax/ops/op_base.py,sha256=W_hiXEW-NxZruQfpMg2W_Aru47HiNZjR0kiEo749XAg,4314
torchax/ops/ops_registry.py,sha256=5wV9BjZqTAUr41tr-A7pchwkQ877YDu_m-RH1f4GWsU,2169
torchax-0.0.9.dist-info/METADATA,sha256=4aW_VaacDUz0sQOcbAPWXl9JUSKv-rGMNHoTh-VIWYg,22302
torchax-0.0.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
torchax-0.0.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
torchax-0.0.9.dist-info/RECORD,,
