torchax 0.0.10.dev20251114__py3-none-any.whl → 0.0.10.dev20251116__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
@@ -22,7 +22,7 @@ from torch.utils import _pytree as pytree
22
22
  from torchax import tensor
23
23
  from contextlib import contextmanager
24
24
 
25
- __version__ = "0.0.10.dev20251114"
25
+ __version__ = "0.0.10.dev20251116"
26
26
  VERSION = __version__
27
27
 
28
28
  # 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.10.dev20251114
3
+ Version: 0.0.10.dev20251116
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=t7SmghL_AbacVWVL3rtr-q8Fw-oZ1M8cKRNbdnrXAxA,4172
2
+ torchax/__init__.py,sha256=V6m7jGoh59hP3nVH7-gK-993qLz5JdmgMIWDbVC3rXw,4172
3
3
  torchax/amp.py,sha256=h5sYm7jhbQhn8kmtdd8ahzrmH06r2VKGYigHQzRzmZc,12366
4
4
  torchax/checkpoint.py,sha256=3nDcGOVbeeBzoaI1dfQxpUPdNiRuGGkVj65DlXkEpF4,2437
5
5
  torchax/config.py,sha256=c2JVtKx-GkkQn9vGxYgweGrm57G60mH1BB-SfE6-d6Q,1497
@@ -25,7 +25,7 @@ torchax/ops/jtorchvision_nms.py,sha256=OZfYntyIatgXtMBU2hywlWvchMKaK028nNsFE2fCy
25
25
  torchax/ops/mappings.py,sha256=RP75HTKJYfZ08dTByfUnCEJRP42jDHcoQXQrBafWI8M,4362
26
26
  torchax/ops/op_base.py,sha256=W_hiXEW-NxZruQfpMg2W_Aru47HiNZjR0kiEo749XAg,4314
27
27
  torchax/ops/ops_registry.py,sha256=5wV9BjZqTAUr41tr-A7pchwkQ877YDu_m-RH1f4GWsU,2169
28
- torchax-0.0.10.dev20251114.dist-info/METADATA,sha256=vVEnW18Y1vesO0NypWIvTtPp8mWNykNl3TG2Vlq6-wI,22315
29
- torchax-0.0.10.dev20251114.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
30
- torchax-0.0.10.dev20251114.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
31
- torchax-0.0.10.dev20251114.dist-info/RECORD,,
28
+ torchax-0.0.10.dev20251116.dist-info/METADATA,sha256=yj4zBuvXI62XXwjDUkMjJgTvvbhWjkCwPIMe4eKS4TQ,22315
29
+ torchax-0.0.10.dev20251116.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
30
+ torchax-0.0.10.dev20251116.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
31
+ torchax-0.0.10.dev20251116.dist-info/RECORD,,