torchax 0.0.11.dev202617__py3-none-any.whl → 0.0.11.dev20251229__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of torchax might be problematic. Click here for more details.

torchax/__init__.py CHANGED
@@ -27,7 +27,7 @@ from torchax import tensor
27
27
 
28
28
  from .checkpoint import load_checkpoint, save_checkpoint
29
29
 
30
- __version__ = "0.0.11.dev202617"
30
+ __version__ = "0.0.11.dev20251229"
31
31
  VERSION = __version__
32
32
 
33
33
  # the "fast path" uses some sparse tensor thingies that currently we
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: torchax
3
- Version: 0.0.11.dev202617
3
+ Version: 0.0.11.dev20251229
4
4
  Summary: torchax is a library for running Jax and PyTorch together
5
5
  Project-URL: Homepage, https://github.com/google/torchax
6
6
  Author-email: Han Qi <qihan.dev@gmail.com>, Google Cloud Inference Team <cmcs-inference-eng@google.com>
@@ -1,5 +1,5 @@
1
1
  torchax/CONTRIBUTING.md,sha256=gbB2ewxDLC-HHRBC3B8HdppV_d9MbDd-9rvzGQt7vZU,1440
2
- torchax/__init__.py,sha256=un4iINEQuZyw0TVP4y6YzmyHoo7yKgDo7eT4_0uXOKY,3889
2
+ torchax/__init__.py,sha256=GYuxucLFbesjExbHu6xu1wjhh1up9ZdzVY8LSwv-EDU,3891
3
3
  torchax/amp.py,sha256=WTrfah2NYodapDVMsht7H3zDpl-XslujxhcYXr85g1s,10999
4
4
  torchax/checkpoint.py,sha256=2eoGeIQtL1Chof0W9qorB2Q0eCyEVJyWKqGoetf32GQ,2439
5
5
  torchax/config.py,sha256=oTwgWDujF9vNSHRNKUvz3ZkocDe0aDaF-yviqAJAewY,1398
@@ -25,7 +25,7 @@ torchax/ops/jtorchvision_nms.py,sha256=VNMshE3LCsIBHVVzkrNEm0kYMF89ZVIKWlQIk6pCZ
25
25
  torchax/ops/mappings.py,sha256=ViEsZaGIi37BhuLw9hx9cA9XXl35OVuaRN6Q9yZygxk,4162
26
26
  torchax/ops/op_base.py,sha256=-rQXLpkgNZ1HM3OT1XQkvAV_7Dtq019_rAXNAg97OuE,4135
27
27
  torchax/ops/ops_registry.py,sha256=sBT41LRGmUVP4ZJ9YU1DyffatOHxe-x8oXqMhCKh0y8,1836
28
- torchax-0.0.11.dev202617.dist-info/METADATA,sha256=mla9SY-2HWZbC0O7cr2TdX4X_bv4fnn6iVruyg-H_BI,22451
29
- torchax-0.0.11.dev202617.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
30
- torchax-0.0.11.dev202617.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
31
- torchax-0.0.11.dev202617.dist-info/RECORD,,
28
+ torchax-0.0.11.dev20251229.dist-info/METADATA,sha256=LhVy_90tMdwogkXLFgi2ljDuCJlMYW94whdboa_khVA,22453
29
+ torchax-0.0.11.dev20251229.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
30
+ torchax-0.0.11.dev20251229.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
31
+ torchax-0.0.11.dev20251229.dist-info/RECORD,,