torchax 0.0.10.dev20251114__py3-none-any.whl → 0.0.11.dev202612__py3-none-any.whl

torchax/view.py

@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import torch
-import torch.utils._pytree as torch_pytree
-import jax
-from enum import Enum
-from typing import Union, List, Tuple, Optional, Any, cast
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
+from enum import Enum
+from typing import Any
+
+import jax
+import torch
 
 # Reference to original PyTorch native functions
 # https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/native_functions.yaml
@@ -37,76 +39,75 @@ class ViewInfoType(Enum):
 
 class ViewInfo(ABC):
   """
-    Abstract base class for all view operations.
-    Defines the interface for applying and updating view transformations.
-    """
+  Abstract base class for all view operations.
+  Defines the interface for applying and updating view transformations.
+  """
 
   def __init__(
-      self,
-      view_info_type: ViewInfoType = ViewInfoType.INVALID,
+    self,
+    view_info_type: ViewInfoType = ViewInfoType.INVALID,
   ):
     """
-        Initialize a ViewInfo object.
+    Initialize a ViewInfo object.
 
-        Args:
-            view_info_type: The type of view operation
-        """
+    Args:
+        view_info_type: The type of view operation
+    """
     self.view_info_type = view_info_type
 
   @abstractmethod
-  def update_tensor(self, new_value: jax.Array,
-                    jax_array: jax.Array) -> jax.Array:
+  def update_tensor(self, new_value: jax.Array, jax_array: jax.Array) -> jax.Array:
     """
-        Apply this view transformation to a JAX array and update its value.
+    Apply this view transformation to a JAX array and update its value.
 
-        Args:
-            new_value: The new values to set in the view
-            jax_array: The parent array to update
+    Args:
+        new_value: The new values to set in the view
+        jax_array: The parent array to update
 
-        Returns:
-            Updated array
-        """
+    Returns:
+        Updated array
+    """
     pass
 
   @abstractmethod
   def transform_tensor(self, jax_array: jax.Array) -> jax.Array:
     """
-        Apply this view transformation to a JAX array.
+    Apply this view transformation to a JAX array.
 
-        Args:
-            jax_array: The array to transform
+    Args:
+        jax_array: The array to transform
 
-        Returns:
-            Transformed array
-        """
+    Returns:
+        Transformed array
+    """
     pass
 
   @abstractmethod
-  def calculate_output_shape(self, source: jax.Array) -> List[int]:
+  def calculate_output_shape(self, source: jax.Array) -> list[int]:
     """
-        Calculate the resulting shape after applying this view.
+    Calculate the resulting shape after applying this view.
 
-        Args:
-            source: Original jax array before transformation
+    Args:
+        source: Original jax array before transformation
 
-        Returns:
-            Resulting shape after transformation
-        """
+    Returns:
+        Resulting shape after transformation
+    """
     pass
 
 
 class NarrowInfo(ViewInfo):
   """
-    Represents a slicing operation on a tensor.
-    Handles operations like tensor[1:3, :, 2:5:2].
-    """
+  Represents a slicing operation on a tensor.
+  Handles operations like tensor[1:3, :, 2:5:2].
+  """
 
-  def __init__(self, slices: Union[slice, Tuple[slice]]) -> None:
+  def __init__(self, slices: slice | tuple[slice]) -> None:
+    """
+    Args:
+        slices: The slice(s) to apply to the tensor.
+            E.g. jax_array.at[slices] will return the transformed tensor.
     """
-        Args:
-            slices: The slice(s) to apply to the tensor.
-                E.g. jax_array.at[slices] will return the transformed tensor.
-        """
     super().__init__(ViewInfoType.NARROW)
     self.slices = slices
 
@@ -121,25 +122,22 @@ class NarrowInfo(ViewInfo):
     except IndexError as e:
       raise IndexError("Invalid slice operation") from e
 
-  def update_tensor(self, new_value: jax.Array,
-                    jax_array: jax.Array) -> jax.Array:
+  def update_tensor(self, new_value: jax.Array, jax_array: jax.Array) -> jax.Array:
     return jax_array.at[self.slices].set(new_value)
 
-  def calculate_output_shape(self, source: jax.Array) -> List[int]:
+  def calculate_output_shape(self, source: jax.Array) -> list[int]:
     return source[self.slices].shape
 
 
 class SelectInfo(ViewInfo):
   """
-    Represents a selection operation on a tensor.
-    Typically used for indexing operations that select specific elements.
-    """
+  Represents a selection operation on a tensor.
+  Typically used for indexing operations that select specific elements.
+  """
 
-  def __init__(self,
-               dim: int = 0,
-               start: int = 0,
-               end: int = 0,
-               stride: int = 0) -> None:
+  def __init__(
+    self, dim: int = 0, start: int = 0, end: int = 0, stride: int = 0
+  ) -> None:
     super().__init__(ViewInfoType.SELECT)
     self.dim: int = dim
     self.start: int = start
@@ -149,29 +147,31 @@ class SelectInfo(ViewInfo):
   def __eq__(self, other: object) -> bool:
     if not isinstance(other, SelectInfo):
       return False
-    return (self.dim == other.dim and self.start == other.start and
-            self.end == other.end and self.stride == other.stride)
+    return (
+      self.dim == other.dim
+      and self.start == other.start
+      and self.end == other.end
+      and self.stride == other.stride
+    )
 
   def transform_tensor(self, jax_array: jax.Array) -> jax.Array:
     raise NotImplementedError("SelectInfo.apply not implemented")
 
-  def update_tensor(self, new_value: jax.Array,
-                    jax_array: jax.Array) -> jax.Array:
+  def update_tensor(self, new_value: jax.Array, jax_array: jax.Array) -> jax.Array:
     raise NotImplementedError("SelectInfo.update not implemented")
 
-  def calculate_output_shape(self, source: jax.Array) -> List[int]:
-    raise NotImplementedError(
-        "SelectInfo.calculate_output_shape not implemented")
+  def calculate_output_shape(self, source: jax.Array) -> list[int]:
+    raise NotImplementedError("SelectInfo.calculate_output_shape not implemented")
 
 
 class AsStridedInfo(ViewInfo):
   """
-    Information for as_strided operations.
-    """
+  Information for as_strided operations.
+  """
 
-  def __init__(self, stride: List[int], offset: int = 0) -> None:
+  def __init__(self, stride: list[int], offset: int = 0) -> None:
     super().__init__(ViewInfoType.AS_STRIDED)
-    self.stride: List[int] = stride
+    self.stride: list[int] = stride
     self.offset: int = offset
 
   def __eq__(self, other: object) -> bool:
@@ -182,28 +182,26 @@ class AsStridedInfo(ViewInfo):
   def transform_tensor(self, jax_array: jax.Array) -> jax.Array:
     raise NotImplementedError("AsStridedInfo.apply not implemented")
 
-  def update_tensor(self, new_value: jax.Array,
-                    jax_array: jax.Array) -> jax.Array:
+  def update_tensor(self, new_value: jax.Array, jax_array: jax.Array) -> jax.Array:
     raise NotImplementedError("AsStridedInfo.update not implemented")
 
-  def calculate_output_shape(self, source: jax.Array) -> List[int]:
-    raise NotImplementedError(
-        "AsStridedInfo.calculate_output_shape not implemented")
+  def calculate_output_shape(self, source: jax.Array) -> list[int]:
+    raise NotImplementedError("AsStridedInfo.calculate_output_shape not implemented")
 
 
 class DiagonalInfo(ViewInfo):
   """
-    Information for diagonal operations.
-    Extracts diagonal elements from a tensor.
-    """
+  Information for diagonal operations.
+  Extracts diagonal elements from a tensor.
+  """
 
   def __init__(self, offset: int = 0, dim1: int = 0, dim2: int = 1) -> None:
     """
-        Args:
-            offset: Offset from the main diagonal
-            dim1: First dimension for diagonal extraction
-            dim2: Second dimension for diagonal extraction
-        """
+    Args:
+        offset: Offset from the main diagonal
+        dim1: First dimension for diagonal extraction
+        dim2: Second dimension for diagonal extraction
+    """
     super().__init__(ViewInfoType.DIAGONAL)
     self.offset: int = offset
     self.dim1: int = dim1
@@ -212,56 +210,65 @@ class DiagonalInfo(ViewInfo):
   def __eq__(self, other: object) -> bool:
     if not isinstance(other, DiagonalInfo):
       return False
-    return (self.offset == other.offset and self.dim1 == other.dim1 and
-            self.dim2 == other.dim2)
+    return (
+      self.offset == other.offset
+      and self.dim1 == other.dim1
+      and self.dim2 == other.dim2
+    )
 
   def transform_tensor(self, jax_array: jax.Array) -> jax.Array:
     raise NotImplementedError("DiagonalInfo.apply not implemented")
 
-  def update_tensor(self, new_value: jax.Array,
-                    jax_array: jax.Array) -> jax.Array:
+  def update_tensor(self, new_value: jax.Array, jax_array: jax.Array) -> jax.Array:
     raise NotImplementedError("DiagonalInfo.update not implemented")
 
-  def calculate_output_shape(self, source: jax.Array) -> List[int]:
-    raise NotImplementedError(
-        "DiagonalInfo.calculate_output_shape not implemented")
+  def calculate_output_shape(self, source: jax.Array) -> list[int]:
+    raise NotImplementedError("DiagonalInfo.calculate_output_shape not implemented")
 
 
 class View(torch.Tensor):
   """
-    A View is a reference to another Tensor or another View,
-    with a transformation applied to it.
-    """
+  A View is a reference to another Tensor or another View,
+  with a transformation applied to it.
+  """
 
   @staticmethod
-  def __new__(cls, parent: Union["torchax.Tensor", "View"], view_info: ViewInfo,
-              env: Any) -> "View":
+  def __new__(
+    cls,
+    parent: torchax.Tensor | View,  # noqa: F821
+    view_info: ViewInfo,
+    env: Any,
+  ) -> View:
+    """
+    Args:
+        parent: Parent tensor or view
+        view_info: Information about the view transformation
+        env: Environment for tensor operations
     """
-        Args:
-            parent: Parent tensor or view
-            view_info: Information about the view transformation
-            env: Environment for tensor operations
-        """
     shape = view_info.calculate_output_shape(parent.jax())
     return torch.Tensor._make_wrapper_subclass(
-        cls,
-        shape,
-        device="meta",
-        dtype=parent.dtype,
-        requires_grad=False,
+      cls,
+      shape,
+      device="meta",
+      dtype=parent.dtype,
+      requires_grad=False,
     )
 
-  def __init__(self, parent: Union["torchax.Tensor", "View"],
-               view_info: ViewInfo, env: Any) -> None:
+  def __init__(
+    self,
+    parent: torchax.Tensor | View,  # noqa: F821
+    view_info: ViewInfo,
+    env: Any,
+  ) -> None:
     super().__init__()
     self.parent = parent
     self.view_info = view_info
     self._env = env
 
-  def get_transformation_chain(self) -> List[ViewInfo]:
+  def get_transformation_chain(self) -> list[ViewInfo]:
+    """
+    Get all view transformations from the source tensor to this view.
     """
-        Get all view transformations from the source tensor to this view.
-        """
     if isinstance(self.parent, View):
       transformations = self.parent.get_transformation_chain()
       transformations.append(self.view_info)
@@ -273,8 +280,8 @@ class View(torch.Tensor):
 
   def source_jax(self) -> jax.Array:
     """
-        Returns the source tensor.
-        """
+    Returns the source tensor.
+    """
     if isinstance(self.parent, View):
       return self.parent.source_jax()
     else:
@@ -282,32 +289,32 @@ class View(torch.Tensor):
 
   def replace_source_jax(self, new_value: jax.Array) -> None:
     """
-        Update the source tensor with new values.
-        """
+    Update the source tensor with new values.
+    """
     if isinstance(self.parent, View):
       self.parent.replace_source_jax(new_value)
     else:
       assert new_value.shape == self.parent._elem.shape
       self.parent._elem = new_value
 
-  def torch(self) -> "torchax.Tensor":
+  def torch(self) -> torchax.Tensor:  # noqa: F821
+    """
+    Returns a Torchax tensor representing this view after all transformations
     """
-        Returns a Torchax tensor representing this view after all transformations
-        """
     from torchax.tensor import Tensor
 
     return Tensor(self.jax(), self._env)
 
   def update(
-      self,
-      new_values: Union[jax.Array, "View", "torchax.Tensor"],
-      view_infos: Optional[List[ViewInfo]] = None,
+    self,
+    new_values: jax.Array | View | torchax.Tensor,  # noqa: F821
+    view_infos: list[ViewInfo] | None = None,
   ) -> None:
     """
-        Update this view with new values, propagating changes back to source.
-        If view_infos is None, it will use the transformation chain
-        from the source tensor.
-        """
+    Update this view with new values, propagating changes back to source.
+    If view_infos is None, it will use the transformation chain
+    from the source tensor.
+    """
     if view_infos is None:
       view_infos = self.get_transformation_chain()
 
@@ -324,14 +331,14 @@ class View(torch.Tensor):
     # And store intermediate values
     intermediate_values = [source_array]
     for view_info in view_infos[:-1]:
-      intermediate_values.append(
-          view_info.transform_tensor(intermediate_values[-1]))
+      intermediate_values.append(view_info.transform_tensor(intermediate_values[-1]))
 
     # TODO: Investigate efficiency of this algorithm
     # Update the source array with the new value by
     # applying inverse transformations in reverse order
     for view_info, parent_array in zip(
-        reversed(view_infos), reversed(intermediate_values)):
+      reversed(view_infos), reversed(intermediate_values), strict=False
+    ):
       # Apply the inverse transformation to propagate changes back
       new_values = view_info.update_tensor(new_values, parent_array)
 
@@ -340,21 +347,22 @@ class View(torch.Tensor):
 
   @classmethod
   def __torch_dispatch__(
-      cls,
-      func: Any,
-      types: Tuple[Any, ...],
-      args: Tuple[Any, ...] = (),
-      kwargs: Optional[dict] = None,
+    cls,
+    func: Any,
+    types: tuple[Any, ...],
+    args: tuple[Any, ...] = (),
+    kwargs: dict | None = None,
   ) -> Any:
     raise AssertionError(
-        'torchax Tensors can only do math within the torchax environment.'
-        'Please wrap your code with `with torchax.default_env()` or '
-        'call torchax.enable_globally() before.')
+      "torchax Tensors can only do math within the torchax environment."
+      "Please wrap your code with `with torchax.default_env()` or "
+      "call torchax.enable_globally() before."
+    )
 
-  def create_sub_view(self, view_info: ViewInfo) -> "View":
+  def create_sub_view(self, view_info: ViewInfo) -> View:
+    """
+    Create a new view that is a child of this view.
     """
-        Create a new view that is a child of this view.
-        """
     return View(self, view_info, self._env)
 
   def __str__(self) -> str:
@@ -362,8 +370,8 @@ class View(torch.Tensor):
 
   def jax(self) -> jax.Array:
     """
-        Returns a copy of the source tensor after transformations.
-        """
+    Returns a copy of the source tensor after transformations.
+    """
     result = self.source_jax()
     for view_info in self.get_transformation_chain():
       result = view_info.transform_tensor(result)

{torchax-0.0.10.dev20251114.dist-info → torchax-0.0.11.dev202612.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torchax
-Version: 0.0.10.dev20251114
+Version: 0.0.11.dev202612
 Summary: torchax is a library for running Jax and PyTorch together
 Project-URL: Homepage, https://github.com/google/torchax
 Author-email: Han Qi <qihan.dev@gmail.com>, Google Cloud Inference Team <cmcs-inference-eng@google.com>
@@ -237,6 +237,12 @@ Description-Content-Type: text/markdown
 
 # torchax: Running PyTorch on TPU via JAX
 
+Docs page: https://google.github.io/torchax/
+Discord Discussion Channel: https://discord.gg/JqeJqGPyzC
+
+
+![](docs/docs/assets/logo.jpeg)
+
 **torchax** is a backend for PyTorch that allows users to run
 PyTorch programs on Google Cloud TPUs. It also provides graph-level
 interoperability between PyTorch and JAX.

torchax-0.0.11.dev202612.dist-info/RECORD

@@ -0,0 +1,31 @@
+torchax/CONTRIBUTING.md,sha256=gbB2ewxDLC-HHRBC3B8HdppV_d9MbDd-9rvzGQt7vZU,1440
+torchax/__init__.py,sha256=m9mLdO6-n-WfpeR-R7jNQxkESUUpXQJ5A-0CNnXuSJs,3889
+torchax/amp.py,sha256=WTrfah2NYodapDVMsht7H3zDpl-XslujxhcYXr85g1s,10999
+torchax/checkpoint.py,sha256=2eoGeIQtL1Chof0W9qorB2Q0eCyEVJyWKqGoetf32GQ,2439
+torchax/config.py,sha256=oTwgWDujF9vNSHRNKUvz3ZkocDe0aDaF-yviqAJAewY,1398
+torchax/decompositions.py,sha256=8VU0FfKqbP8h3S7JzHi0iWqT5E1OrwYvuhf6cjzDTlI,29303
+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=6R9_pIkd5Kwb5EoACUsq4GEEe8PaSFOdfnD4GIy6Y9U,11815
+torchax/mesh_util.py,sha256=Y3RVKOyLVKpbseyXTYlJlUgyNavmDAsh2MQ6pXYQDUU,9719
+torchax/tensor.py,sha256=IP7JuzUfKZVrhJdA67VdXBEh4bfypwavKRzw6llsD4U,21452
+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=x5vrEiVRFaTigPYvFHtR9ClwW0Z6TEHT66-mEdzfe3k,163435
+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=OuFTed92B3WuWnfnjBBvMgwkl43PLJJiDWSMb0ZS0sw,16406
+torchax/ops/jtorchvision_nms.py,sha256=VNMshE3LCsIBHVVzkrNEm0kYMF89ZVIKWlQIk6pCZB0,9197
+torchax/ops/mappings.py,sha256=ViEsZaGIi37BhuLw9hx9cA9XXl35OVuaRN6Q9yZygxk,4162
+torchax/ops/op_base.py,sha256=-rQXLpkgNZ1HM3OT1XQkvAV_7Dtq019_rAXNAg97OuE,4135
+torchax/ops/ops_registry.py,sha256=sBT41LRGmUVP4ZJ9YU1DyffatOHxe-x8oXqMhCKh0y8,1836
+torchax-0.0.11.dev202612.dist-info/METADATA,sha256=tVMhNPH26W_cwlb9oA8xPNByKjM25aY7OI4SGnGFTg4,22451
+torchax-0.0.11.dev202612.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+torchax-0.0.11.dev202612.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+torchax-0.0.11.dev202612.dist-info/RECORD,,

{torchax-0.0.10.dev20251114.dist-info → torchax-0.0.11.dev202612.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any

torchax-0.0.10.dev20251114.dist-info/RECORD

@@ -1,31 +0,0 @@
-torchax/CONTRIBUTING.md,sha256=gbB2ewxDLC-HHRBC3B8HdppV_d9MbDd-9rvzGQt7vZU,1440
-torchax/__init__.py,sha256=t7SmghL_AbacVWVL3rtr-q8Fw-oZ1M8cKRNbdnrXAxA,4172
-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=GEc8s0GjCF6N9A8UnIyLmtch5TwCgnqWbTRr5iMerkk,177058
-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.10.dev20251114.dist-info/METADATA,sha256=vVEnW18Y1vesO0NypWIvTtPp8mWNykNl3TG2Vlq6-wI,22315
-torchax-0.0.10.dev20251114.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-torchax-0.0.10.dev20251114.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-torchax-0.0.10.dev20251114.dist-info/RECORD,,