torchax/__init__.py,sha256=O6eXbqeXEXJgHI4VbYVudTh2Gbu7JNOiYjmiQLAVSjY,4895
torchax/_torch_compat.py,sha256=N-hr1oTrwp-jYplTAZgLn4Ky5qIXUgwJaJtXHA1F_lk,826
torchax/amp.py,sha256=psymWxP3tIfpAu9OK-pCldEr2Skmifh0M3OOzX5ftRU,11347
torchax/checkpoint.py,sha256=V0Bb_J5loX5a3A4beh0QXqI-MH1kO8Y1P-zG6Go130I,2669
torchax/config.py,sha256=oTwgWDujF9vNSHRNKUvz3ZkocDe0aDaF-yviqAJAewY,1398
torchax/decompositions.py,sha256=TehSQf6B94u_1yehUkZf49ioNc0sKHd1aOMY0FdIbAo,28486
torchax/device_module.py,sha256=7WrLUBjMQiAilVfRwEwJrbfkizPZAC3022UO70U5uEQ,924
torchax/export.py,sha256=vZ5NpvPUld6oo0JdVtY1hfob6Ll1ruoLmnfpJ5B2VEc,20183
torchax/flax.py,sha256=Eft46Np3qPvSLmBGOltCx8KbnrGsuBeqx3Zu0tlhMpg,1807
torchax/interop.py,sha256=LZ6sHhBGb5__NmhtMdIJUz4etZajBuCzD_uBpxzo5BY,15413
torchax/mesh_util.py,sha256=Y3RVKOyLVKpbseyXTYlJlUgyNavmDAsh2MQ6pXYQDUU,9719
torchax/tensor.py,sha256=2CNncaPP-YlJleaznd6cixnKOakxNidZN3ce33gBgXM,22651
torchax/train.py,sha256=3sqIYO1Q6GN6gRGkVVoKjjUZ3xYPgh110g5XbkaUxh4,4367
torchax/types.py,sha256=NDtW1fARypg0ZHcGRVTBZKQqxwJzWtf5E5bntng91gk,981
torchax/util.py,sha256=Oud7oaAw1SJo_v4fwEZdjuseZ_bvngAsAQ-dOEzy_20,3675
torchax/view.py,sha256=750VYe6tmwAINNAyjN8GDvPmgaR8luvr74szwuikGts,11256
torchax/ops/__init__.py,sha256=uc-Rod4Xlk_oJ-rdY9P8-MGTu9dsXPombscqcSys0r8,840
torchax/ops/jaten.py,sha256=txVVtEPrrD4pnecqfKFM6Ugdci30FoI9VZZfLnMpxm4,167809
torchax/ops/jax_reimplement.py,sha256=Te8Je2ea9jX2SFV34PNPSHVP7-z_bmFykVWeqn8Tqwo,7714
torchax/ops/jc10d.py,sha256=sO99kYDM9WRnSENHmMkH_MXWkx6HZdDvK5Z_M0C571g,1889
torchax/ops/jimage.py,sha256=uvYW-fMaGU6-QTXAmTZ8rmHEkkwpXh2I-bu54STf-ic,3594
torchax/ops/jlibrary.py,sha256=ESeFS5xwV1yiCIp_yKzDXihHd1wcz5eXjFkjFKsmw3w,3470
torchax/ops/jtorch.py,sha256=MPAWgPW25y3iBc4EfiQghZI5jA8Uf2TbBLUcAZU71-s,17389
torchax/ops/jtorchvision_nms.py,sha256=VNMshE3LCsIBHVVzkrNEm0kYMF89ZVIKWlQIk6pCZB0,9197
torchax/ops/mappings.py,sha256=ViEsZaGIi37BhuLw9hx9cA9XXl35OVuaRN6Q9yZygxk,4162
torchax/ops/op_base.py,sha256=9ojGHGmNh6jhusvXVOAp95UZQfYs7rzozBxEvf9jRZ4,4366
torchax/ops/ops_registry.py,sha256=sBT41LRGmUVP4ZJ9YU1DyffatOHxe-x8oXqMhCKh0y8,1836
torchax-0.0.14.dev20260913.dist-info/METADATA,sha256=VIlQjhCgSPnuMLQzQtM8n-HllHWGan1g8oQ8nny0k9g,22453
torchax-0.0.14.dev20260913.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
torchax-0.0.14.dev20260913.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
torchax-0.0.14.dev20260913.dist-info/RECORD,,
