torchax/__init__.py,sha256=TmjmyQ43bD_pcgsc07D08HjqCnQWz6Wl9MJZlapNpjg,4895
torchax/_torch_compat.py,sha256=N-hr1oTrwp-jYplTAZgLn4Ky5qIXUgwJaJtXHA1F_lk,826
torchax/amp.py,sha256=sPqtzvapmD4WKB6pbZ5q5REZQc9JpnziqjKtP_On4AI,11253
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=CFQESWy9-ENo9Ozf08qwqUtrNpt7Y3ubX_dP99o3FdM,9395
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=EXRZRtayyDUWwjGTrAx8PFREtAVhQM1oN62Uyz8Ohwo,167762
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.dev20260809.dist-info/METADATA,sha256=AYJObtVLJgpRE1ADEu2Mr_RwfSUbHCTjNx6ejA0qMck,22453
torchax-0.0.14.dev20260809.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
torchax-0.0.14.dev20260809.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
torchax-0.0.14.dev20260809.dist-info/RECORD,,
