ai-edge-torch-nightly 0.7.0.dev20250929__py3-none-any.whl → 0.8.0.dev20251206__py3-none-any.whl

ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py

@@ -21,6 +21,7 @@ from ai_edge_torch.odml_torch.lowerings import registry
 import jax
 import jax.numpy as jnp
 from jax._src.lib.mlir import ir
+import numpy as np
 import torch
 import torch_xla2.ops.jaten  # Import to load torch_xla2 ops
 import torch_xla2.ops.ops_registry  # Import to load torch_xla2 ops
@@ -71,8 +72,6 @@ lower_by_torch_xla2(torch.ops.aten._cdist_forward)
 lower_by_torch_xla2(torch.ops.aten._local_scalar_dense)
 lower_by_torch_xla2(torch.ops.aten._local_scalar_dense)
 lower_by_torch_xla2(torch.ops.aten._log_softmax)
-lower_by_torch_xla2(torch.ops.aten._native_batch_norm_legit)
-lower_by_torch_xla2(torch.ops.aten._native_batch_norm_legit_no_training)
 lower_by_torch_xla2(torch.ops.aten._pdist_forward)
 lower_by_torch_xla2(torch.ops.aten._softmax)
 lower_by_torch_xla2(torch.ops.aten._unsafe_index)
@@ -158,10 +157,8 @@ lower_by_torch_xla2(torch.ops.aten.logical_not)
 lower_by_torch_xla2(torch.ops.aten.logical_or)
 lower_by_torch_xla2(torch.ops.aten.logical_xor)
 lower_by_torch_xla2(torch.ops.aten.max)
-lower_by_torch_xla2(torch.ops.aten.max_pool2d_with_indices)
 lower_by_torch_xla2(torch.ops.aten.max_pool2d_with_indices_backward)
 lower_by_torch_xla2(torch.ops.aten.max_pool2d_with_indices_backward)
-lower_by_torch_xla2(torch.ops.aten.max_pool3d_with_indices)
 lower_by_torch_xla2(torch.ops.aten.maximum)
 lower_by_torch_xla2(torch.ops.aten.mean)
 lower_by_torch_xla2(torch.ops.aten.min)
@@ -175,7 +172,6 @@ lower_by_torch_xla2(torch.ops.aten.nonzero)
 lower_by_torch_xla2(torch.ops.aten.outer)
 lower_by_torch_xla2(torch.ops.aten.permute)
 lower_by_torch_xla2(torch.ops.aten.permute_copy)
-lower_by_torch_xla2(torch.ops.aten.pixel_shuffle)
 lower_by_torch_xla2(torch.ops.aten.pow)
 lower_by_torch_xla2(torch.ops.aten.prod)
 lower_by_torch_xla2(torch.ops.aten.reciprocal)
@@ -240,6 +236,249 @@ lower_by_torch_xla2(torch.ops.prims.broadcast_in_dim)
 lower_by_torch_xla2(torch.ops.prims.var)
 
 
+def _ceil_mode_padding(
+    padding: list[int],
+    input_shape: list[int],
+    kernel_size: list[int],
+    stride: list[int],
+    dilation: list[int],
+    ceil_mode: bool,
+):
+  """Creates low and high padding specification for ceil mode.
+
+  This is for the given padding (which is symmetric). Additional high padding
+  could be required when ceil mode is set.
+  """
+  ceil_mode_padding = []
+  for i in range(len(padding)):
+    left_padding = padding[i]
+    right_padding = left_padding
+
+    input_size = input_shape[2 + i]
+    output_size_rem = (
+        input_size + 2 * left_padding - (kernel_size[i] - 1) * dilation[i] - 1
+    ) % stride[i]
+    if ceil_mode and output_size_rem != 0:
+      extra_padding = stride[i] - output_size_rem
+      new_output_size = (
+          input_size
+          + left_padding
+          + right_padding
+          + extra_padding
+          - (kernel_size[i] - 1) * dilation[i]
+          - 1
+          + stride[i]
+          - 1
+      ) // stride[i] + 1
+      # Ensure that the last pooling starts inside the image.
+      size_to_compare = input_size + left_padding
+
+      if (new_output_size - 1) * stride[i] < size_to_compare:
+        right_padding += extra_padding
+
+    ceil_mode_padding.append((left_padding, right_padding))
+  return ceil_mode_padding
+
+
+def max_pool(
+    inputs,
+    kernel_size,
+    strides=None,
+    padding=0,
+    dilation=1,
+    ceil_mode=False,
+    with_index=False,
+):
+  num_spatial_dims = len(kernel_size)
+  num_batch_dims = inputs.ndim - num_spatial_dims - 1
+  kernel_size_tup = tuple(kernel_size)
+  # Default stride is kernel_size
+  strides_tup = tuple(strides) if strides else kernel_size_tup
+  if isinstance(padding, int):
+    padding_list = [padding for _ in range(num_spatial_dims)]
+  elif not padding:  # padding can be [], meaning all zeros.
+    padding_list = [0 for _ in range(num_spatial_dims)]
+  else:
+    padding_list = padding
+
+  if isinstance(dilation, int):
+    dilation_tup = tuple(dilation for _ in range(num_spatial_dims))
+  elif not dilation:
+    dilation_tup = tuple(1 for _ in range(num_spatial_dims))
+  elif isinstance(dilation, list):
+    dilation_tup = tuple(dilation)
+  else:
+    dilation_tup = dilation
+
+  input_shape_for_ceil = inputs.shape
+  if num_batch_dims == 0:
+    input_shape_for_ceil = [1, *input_shape_for_ceil]
+  padding_pairs = _ceil_mode_padding(
+      padding_list,
+      input_shape_for_ceil,
+      kernel_size_tup,
+      strides_tup,
+      dilation_tup,
+      ceil_mode,
+  )
+
+  assert len(kernel_size_tup) == len(
+      strides_tup
+  ), f"len({kernel_size_tup=}) must equal len({strides_tup=})"
+  assert len(kernel_size_tup) == len(
+      dilation_tup
+  ), f"len({kernel_size_tup=}) must equal len({dilation_tup=})"
+
+  is_single_input = False
+  if num_batch_dims == 0:
+    inputs = inputs[None]
+    is_single_input = True
+
+  reduce_window_strides = (1,) * (inputs.ndim - num_spatial_dims) + strides_tup
+  reduce_window_dims = (1,) * (inputs.ndim - num_spatial_dims) + kernel_size_tup
+  reduce_window_dilation = (1,) * (
+      inputs.ndim - num_spatial_dims
+  ) + dilation_tup
+
+  assert inputs.ndim == len(
+      reduce_window_dims
+  ), f"len({inputs.shape}) != len({reduce_window_dims})"
+  if not isinstance(padding_pairs, str):
+    padding_pairs_tup = tuple(padding_pairs)
+    assert all(
+        [len(x) == 2 for x in padding_pairs_tup]
+    ), f"each entry in padding {padding_pairs_tup} must be length 2"
+    padding_lax = ((0, 0),) * (
+        inputs.ndim - len(padding_pairs_tup)
+    ) + padding_pairs_tup
+  else:
+    padding_lax = padding_pairs
+
+  indices = jnp.arange(
+      np.prod(inputs.shape[-num_spatial_dims:]), dtype=jnp.int64
+  )
+  indices = indices.reshape(inputs.shape[-num_spatial_dims:])
+  indices_shape = (1,) * (inputs.ndim - indices.ndim) + indices.shape
+  indices = jnp.broadcast_to(indices.reshape(indices_shape), inputs.shape)
+
+  return_dtype = inputs.dtype
+  if jnp.issubdtype(inputs.dtype, jnp.integer):
+    init_val = jnp.int32(jnp.iinfo(jnp.int32).min)
+    inputs = inputs.astype(jnp.int32)
+  else:
+    init_val = jnp.float32(-jnp.inf)
+    inputs = inputs.astype(jnp.float32)
+
+  if not with_index:
+    y = jax.lax.reduce_window(
+        inputs,
+        init_val,
+        jax.lax.max,
+        reduce_window_dims,
+        reduce_window_strides,
+        padding_lax,
+        window_dilation=reduce_window_dilation,
+    )
+    if is_single_input:
+      y = jnp.squeeze(y, axis=0)
+    return y.astype(return_dtype)
+  else:
+
+    def reduce_fn(a, b):
+      ai, av = a
+      bi, bv = b
+      which = av >= bv
+      return jnp.where(which, ai, bi), jnp.where(which, av, bv)
+
+    indices, y = jax.lax.reduce_window(
+        (indices, inputs),
+        (jnp.int64(0), init_val),
+        reduce_fn,
+        reduce_window_dims,
+        reduce_window_strides,
+        padding_lax,
+        window_dilation=reduce_window_dilation,
+    )
+    if is_single_input:
+      indices = jnp.squeeze(indices, axis=0)
+      y = jnp.squeeze(y, axis=0)
+    y = y.astype(return_dtype)
+    return y, indices
+
+
+@lower_by_jax(torch.ops.aten.max_pool2d_with_indices)
+def _aten_max_pool2d_with_indices(
+    self, kernel_size, stride=None, padding=0, dilation=1, ceil_mode=False
+):
+  stride = stride if stride is not None else []
+  y = max_pool(
+      self,
+      kernel_size,
+      strides=stride,
+      padding=padding,
+      dilation=dilation,
+      ceil_mode=ceil_mode,
+      with_index=False,
+  )
+  # TFLite's reduce_window kernel doesn't support multiple inputs/outputs,
+  # so we emit reduce_window with a single output and return dummy indices.
+  return y, jnp.zeros_like(y, dtype=jnp.int64)
+
+
+@lower_by_jax(torch.ops.aten.max_pool3d_with_indices.default)
+def _aten_max_pool3d_with_indices(
+    self, kernel_size, stride=None, padding=0, dilation=1, ceil_mode=False
+):
+  stride = stride if stride is not None else []
+  y = max_pool(
+      self,
+      kernel_size,
+      strides=stride,
+      padding=padding,
+      dilation=dilation,
+      ceil_mode=ceil_mode,
+      with_index=False,
+  )
+  # TFLite's reduce_window kernel doesn't support multiple inputs/outputs,
+  # so we emit reduce_window with a single output and return dummy indices.
+  return y, jnp.zeros_like(y, dtype=jnp.int64)
+
+
+@lower_by_jax(torch.ops.aten.pixel_shuffle)
+def _aten_pixel_shuffle(x, upscale_factor):
+  """PixelShuffle implementation in JAX lowering.
+
+  Args:
+    x: Input tensor. Typically a feature map.
+    upscale_factor: Integer by which to upscale the spatial dimensions.
+
+  Returns:
+    Tensor after PixelShuffle operation.
+  """
+
+  batch_size, channels, height, width = x.shape
+
+  if channels % (upscale_factor**2) != 0:
+    raise ValueError(
+        "Number of channels must be divisible by the square of the upscale"
+        " factor."
+    )
+
+  new_channels = channels // (upscale_factor**2)
+  new_height = height * upscale_factor
+  new_width = width * upscale_factor
+
+  x = x.reshape(
+      batch_size, new_channels, upscale_factor, upscale_factor, height, width
+  )
+  x = jnp.transpose(
+      x, (0, 1, 4, 2, 5, 3)
+  )  # Move channels to spatial dimensions
+  x = x.reshape(batch_size, new_channels, new_height, new_width)
+
+  return x
+
+
 @lower_by_jax(torch.ops.aten.unbind)
 def _aten_copy(self, *args, **kwargs):
   return _TORCH_XLA2_IMPLS[torch.ops.aten.unbind_copy](self, *args, **kwargs)
@@ -250,6 +489,17 @@ def _aten_copy(self, src, **kwargs):
   return _TORCH_XLA2_IMPLS[torch.ops.aten.copy](self, src)
 
 
+@lower_by_jax(torch.ops.aten.elu.default)
+def _aten_elu(self, alpha=1.0, scale=1.0, input_scale=1.0):
+  pos_coef = scale
+  neg_coef = alpha * scale
+  neg_input_coef = input_scale
+
+  pos_branch = self * pos_coef
+  neg_branch = jnp.expm1(self * neg_input_coef) * neg_coef
+  return jnp.where(self >= 0, pos_branch, neg_branch)
+
+
 @registry.lower(torch.ops.aten.add.Scalar)
 def _aten_add_scalar(lctx: LoweringContext, self, other):
   _log_usage(torch.ops.aten.add.Scalar)

ai_edge_torch/version.py

@@ -15,4 +15,4 @@
 
 # The next version of ai-edge-torch.
 # The minor version code should be bumped after every release.
-__version__ = "0.7.0.dev20250929"
+__version__ = "0.8.0.dev20251206"

{ai_edge_torch_nightly-0.7.0.dev20250929.dist-info → ai_edge_torch_nightly-0.8.0.dev20251206.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: ai-edge-torch-nightly
-Version: 0.7.0.dev20250929
+Version: 0.8.0.dev20251206
 Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
 Home-page: https://github.com/google-ai-edge/ai-edge-torch
 Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI
@@ -13,6 +13,8 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: Mathematics
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -37,7 +39,17 @@ Requires-Dist: ai-edge-quantizer-nightly
 Requires-Dist: jax
 Requires-Dist: torch-xla2[odml]>=0.0.1.dev20241201
 Provides-Extra: torch-xla
-Requires-Dist: torch-xla>=2.4.0; extra == "torch-xla"
+Requires-Dist: torch_xla>=2.4.0; extra == "torch-xla"
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 Library that supports converting PyTorch models into a .tflite format, which can
 then be run with TensorFlow Lite and MediaPipe.  This enables applications for

{ai_edge_torch_nightly-0.7.0.dev20250929.dist-info → ai_edge_torch_nightly-0.8.0.dev20251206.dist-info}/RECORD

@@ -2,9 +2,9 @@ ai_edge_torch/__init__.py,sha256=lemyLCNoGYRnJsmDuGZu7qOqLbLqG6CGDFtu3ue1syU,129
 ai_edge_torch/_config.py,sha256=AiqhbcheF7j_ozIGDLC89k1we95aVgFDa-tR6h7UI0s,2529
 ai_edge_torch/conftest.py,sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI,758
 ai_edge_torch/model.py,sha256=A7loFu8jE9CsXsfMmHYZ-KDFJiaD8Kkqwm_9d3IVzk0,5638
-ai_edge_torch/version.py,sha256=Kfpfj3j3af9tnh8DVfa1oaj9-YZAfki1hWVm44kYPfU,806
+ai_edge_torch/version.py,sha256=igYuweKQrisARFhIhK3eg0wnxeYvQtBYWgM2qt2tV7M,806
 ai_edge_torch/_convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/_convert/conversion.py,sha256=iQk3R-pLq4c1nfLqPB4xTRj78gghxPGzJCJtILLdg5o,6123
+ai_edge_torch/_convert/conversion.py,sha256=JqGZZGbpTmYiT-ta07IQbJ9-gFm-3Vip2aSzW9ulIng,6117
 ai_edge_torch/_convert/conversion_utils.py,sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o,2160
 ai_edge_torch/_convert/converter.py,sha256=6MLKELzAwFoiXv-b7KRYi7gc7Z57XOeowcz9ArIl9TM,12100
 ai_edge_torch/_convert/signature.py,sha256=-YKJdLk-eNEHfhdPCtcQVtZf915SoVePEFxKXPPf16c,2572
@@ -43,7 +43,7 @@ ai_edge_torch/examples/selfie_segmentation/model.py,sha256=5otCH1MzNgSP0fikYq53h
 ai_edge_torch/experimental/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/fx_infra/__init__.py,sha256=APjkSqEfwDxcnI8k53rGi3Ef-G2L-M8fdaPGpxXtuiI,1347
 ai_edge_torch/fx_infra/_canonicalize_pass.py,sha256=GDRoDdPVQw--QQFTT5J_C3TVuphL31m6K6F1-67SE4s,1097
-ai_edge_torch/fx_infra/_safe_run_decompositions.py,sha256=ZbWheeZ8ydsxCk2aVGUgUynrkEkBOMjBCzPhS5uq4sU,2595
+ai_edge_torch/fx_infra/_safe_run_decompositions.py,sha256=V-vhvScNrE3nTenT1LIULbWIU-2FD_OHthFZCPuxtzk,3480
 ai_edge_torch/fx_infra/decomp.py,sha256=S58SCgwMHYVFl_hJwlJxvu2wcI-AGNn82gel3qmTPrU,2500
 ai_edge_torch/fx_infra/graph_utils.py,sha256=nqGe-xIJ77RamSUh0UYyI2XHOsZqFDWax-vpRAtVR_E,2796
 ai_edge_torch/fx_infra/pass_base.py,sha256=Ic2AlhSoRFscz6l7gJKvWVNMDLQFfAw5kRf84-ZR9qM,2904
@@ -54,36 +54,36 @@ ai_edge_torch/generative/custom_ops/dynamic_update_slice.py,sha256=ZGAq2CfWZsfef
 ai_edge_torch/generative/examples/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/amd_llama_135m/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/amd_llama_135m/amd_llama_135m.py,sha256=Z2oUQ--pXPgNQyGLdWDNdyC90cSZzDN9YnSclRrPWAg,2703
-ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py,sha256=ze0MPZsUJpIBSCMJB4BrnJT5MFomRcffvuZEqKG20AM,1848
+ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py,sha256=Mk_wGuQmdET3VCxD-QbAyDVSj8J5pKwS4Nq1jZ1gu2Y,1109
 ai_edge_torch/generative/examples/amd_llama_135m/verify.py,sha256=uyBg5-trxQEjEHDZMX4qojkcsZgERUiPqIgR9n0_AY4,1311
 ai_edge_torch/generative/examples/amd_llama_135m/verify_util.py,sha256=J_FwH5KzS1tlyxcICSPir35gP2tWqRQ5wyVTsV9Uk8w,3018
 ai_edge_torch/generative/examples/deepseek/__init__.py,sha256=JaAnrFoXTl3RJX97XspklkTyqOHVyAgRJsZtzNDd10c,671
-ai_edge_torch/generative/examples/deepseek/convert_to_tflite.py,sha256=wcAk6EgNWXzYx_vQCmMDAAYw398ztlUdk_Gj9CYTsOg,1883
+ai_edge_torch/generative/examples/deepseek/convert_to_tflite.py,sha256=PtaBKoq-vocegiBJGk9GpHGvCreDu4WBFxsakSZgnA8,1144
 ai_edge_torch/generative/examples/deepseek/deepseek.py,sha256=cm3rLw6aHFWQswwCqfvwxr05JG0maSzNVn1pBqD19qY,2816
 ai_edge_torch/generative/examples/deepseek/verify.py,sha256=HkvgEyGb-V_f6mWfyeN7Ai5uADAVQNzWvkygaKJiLAc,1344
 ai_edge_torch/generative/examples/deepseek/verify_util.py,sha256=xf2-jhv-ORwr_-Rs2UBpO2vcAwJNZIdmOxpcuKIQaww,3028
 ai_edge_torch/generative/examples/gemma/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/gemma/convert_gemma1_to_tflite.py,sha256=m5N3MxNX4Yu8-0vXRszkMgfVuFKN6gskVoUIGqonJFk,1814
-ai_edge_torch/generative/examples/gemma/convert_gemma2_to_tflite.py,sha256=fR4869w1UZIuUVGLDdPof0IDEMDhqn2ej5kl7QW9vNA,1882
+ai_edge_torch/generative/examples/gemma/convert_gemma1_to_tflite.py,sha256=jJQ43cmCu4Ys-c0WWPkJSA5U-5FgpZ30JOPWF5QUiWg,1075
+ai_edge_torch/generative/examples/gemma/convert_gemma2_to_tflite.py,sha256=LlxT2fE8Sq6Cg6YY8c0HnSc-T8831xeJsMKBcKVk77Y,1143
 ai_edge_torch/generative/examples/gemma/gemma1.py,sha256=8oLB0PaSrHMy0tDT--qPIEjTm0qNT16kqRBCHqAALSc,4674
 ai_edge_torch/generative/examples/gemma/gemma2.py,sha256=E6jotWYYIx6SXUWqurKWjiZpbfj_M2jJrBc2rQ90z1s,11782
 ai_edge_torch/generative/examples/gemma/verify_gemma1.py,sha256=USyist332qZkhCBof2tJwQSqtnKjTQsKAK_jCE_CO2U,1853
 ai_edge_torch/generative/examples/gemma/verify_gemma2.py,sha256=eAM7EVVMW-QCqjeZEss7TOkVKArgUs1La51LAC-5a9A,1962
 ai_edge_torch/generative/examples/gemma/verify_util.py,sha256=977W4szm0gklhsVduiqHpkgc4QjpMn_IkRRvaqQC7ac,8422
 ai_edge_torch/generative/examples/gemma3/__init__.py,sha256=JaAnrFoXTl3RJX97XspklkTyqOHVyAgRJsZtzNDd10c,671
-ai_edge_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py,sha256=b2l8R3-3xhurmXn3LRZ-C12J11C8tx77NXKze6jFYbA,2425
+ai_edge_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py,sha256=9oNL1rSvh4CHWsIlK-_n6FkNaPv9B1tWTXf-IBPk2jA,1475
 ai_edge_torch/generative/examples/gemma3/decoder.py,sha256=DEkOnfHZJPaxbjwegarRjGpBa7gr6IuWMf8xgZP2mtU,17951
 ai_edge_torch/generative/examples/gemma3/gemma3.py,sha256=PAtuqvBkq6TeNFImoJfCaLX5pZCBOcaQjFvwur_XQJ4,7141
 ai_edge_torch/generative/examples/gemma3/image_encoder.py,sha256=OCMIAQfNmPR4uQUAtlYL6j4xkG0dw2Ays4-lnThcWqQ,5110
 ai_edge_torch/generative/examples/gemma3/verify_gemma3.py,sha256=v8oNXFICmVOtQxfO7IhZ8GnbvotEkDi9lzYHjoQyOso,2464
 ai_edge_torch/generative/examples/gemma3/verify_util.py,sha256=5OmUwz38kVHYLA-v8U8evvDN9da2WioZtGo-XK6yq1o,10067
 ai_edge_torch/generative/examples/hammer/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/hammer/convert_to_tflite.py,sha256=BujBJml_ShWpKJyUgKAJqV4C6yd1uODjKg8fOgchPKI,2041
+ai_edge_torch/generative/examples/hammer/convert_to_tflite.py,sha256=OudSFnxLJrQ89vUl4yFddcMS22UguYff2dmvheX_-MI,1302
 ai_edge_torch/generative/examples/hammer/hammer.py,sha256=0fz6LmZLXrVyty6w5T6LWRoqvQ-5XYREH6gFzB_gOu0,3737
 ai_edge_torch/generative/examples/hammer/verify.py,sha256=iuKFMkI1VZc9_ESUQr32cHWHsDtYOAF702TdqhD3Ns4,1589
 ai_edge_torch/generative/examples/hammer/verify_util.py,sha256=vB5kiFcDBM2G-AkTByJarxur-kJT_J2wGCW4mzf5whc,2962
 ai_edge_torch/generative/examples/llama/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/llama/convert_to_tflite.py,sha256=K-hke37VBMWakuqCeRmpaZBujGWRykDIaUBKJQ0nI5g,2022
+ai_edge_torch/generative/examples/llama/convert_to_tflite.py,sha256=6or963hGkTmvfxhpYqNY--1aPIjPVM-V_WjhWanoiFU,1283
 ai_edge_torch/generative/examples/llama/llama.py,sha256=PpC3idSFSATjQzcLv8oXwo66_5zU-B_np2YVAs2Naog,6602
 ai_edge_torch/generative/examples/llama/verify.py,sha256=XoF_-kxdryjt0Bt_YeHnIbLfjwFxSVioTSEG75moDr8,1581
 ai_edge_torch/generative/examples/llama/verify_util.py,sha256=YVwoiQNwnwc_yx2ra3KYMCmqY-PiWm1kYZQjcw2BXws,3076
@@ -91,7 +91,7 @@ ai_edge_torch/generative/examples/moonshine/__init__.py,sha256=hHLluseD2R0Hh4W6X
 ai_edge_torch/generative/examples/moonshine/convert_moonshine_to_tflite.py,sha256=_GkaSkregS3NWN38UGXxj4pED5gtQGaaPZx5_CZ0TVM,1657
 ai_edge_torch/generative/examples/moonshine/moonshine.py,sha256=nZ2b8u4TmsB5sgdClgAuH8E78bcTv9RCnF9666HqP2M,3394
 ai_edge_torch/generative/examples/openelm/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/openelm/convert_to_tflite.py,sha256=C5KEMMle8v6yZWrJPYoN-1CAjeZv7GQEJvyL28DeEYs,1813
+ai_edge_torch/generative/examples/openelm/convert_to_tflite.py,sha256=TAfVisvM8kcRQs2p6K_09pYN-dySdn8toS49CnYdZyE,1074
 ai_edge_torch/generative/examples/openelm/openelm.py,sha256=Zfoe0VzNHiCK3CIlp0UYzxxJuE9yZXrSCb84S24LMM0,4448
 ai_edge_torch/generative/examples/openelm/verify.py,sha256=kRoNEUEsrz51PFSeTPcrYsPBQRLtUmYM3t_-Jl0oFqM,1300
 ai_edge_torch/generative/examples/openelm/verify_util.py,sha256=3Tay96oEiV3Gtz-K2Gu2-tov2Y4mMpztPhwXfs4c5EA,2970
@@ -106,9 +106,9 @@ ai_edge_torch/generative/examples/paligemma/verify_decoder.py,sha256=Z-Xx2mpygBc
 ai_edge_torch/generative/examples/paligemma/verify_decoder2.py,sha256=ryn3OpMFeVwUXoywl73XyypivjghSBojdLDLzvF3XI4,2595
 ai_edge_torch/generative/examples/paligemma/verify_image_encoder.py,sha256=vNm-wTT8BD6zbX6GocfP1QrVoHl0zSvuVxoXN36eeiU,3540
 ai_edge_torch/generative/examples/phi/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/phi/convert_phi3_to_tflite.py,sha256=th7EYY2NmT4qLMtQsEIU697sGhwZi24V8OwB6xZRHEw,1802
-ai_edge_torch/generative/examples/phi/convert_phi4_to_tflite.py,sha256=OkqJXktzNb8VO4_gNSl1iKIXBDMx0McCOxnoEUrq_B4,1800
-ai_edge_torch/generative/examples/phi/convert_to_tflite.py,sha256=Ff0A2NOpcmwGi87AWdQ4X2ZXeALQVzVvAGdnBzcZgVY,1801
+ai_edge_torch/generative/examples/phi/convert_phi3_to_tflite.py,sha256=X5j8Gm6BsgR26evL1JqJlGj_-ezYaICKm4Bn131MX-o,1063
+ai_edge_torch/generative/examples/phi/convert_phi4_to_tflite.py,sha256=N6-NpAr3pvKlxyzeyhC_p84V3OpdaLlCwWeu3mCLRIw,1061
+ai_edge_torch/generative/examples/phi/convert_to_tflite.py,sha256=uM4dYRkMJpYg_fxQCLhLTNPKM5KIMPCf7yH2z6AD1q4,1062
 ai_edge_torch/generative/examples/phi/phi2.py,sha256=CyqXhCcegY7mcsuRBg8-aXvOHTK4PZufGKBWb0mgcNE,3361
 ai_edge_torch/generative/examples/phi/phi3.py,sha256=lO9vVCTqtbVZuSjJGRlmYrh5ZmLPfrqg7RXgTsqtggg,6811
 ai_edge_torch/generative/examples/phi/phi4.py,sha256=n-LLniR-Q3Nyz4JCTvKZe9hBHVWBGtIBQdqe7_6xAyM,5596
@@ -117,8 +117,8 @@ ai_edge_torch/generative/examples/phi/verify_phi3.py,sha256=TwIu2xUPQyMUTFdz29E2
 ai_edge_torch/generative/examples/phi/verify_phi4.py,sha256=2MlgQrfRkhE7Dya8MIixGwpqEZYdPjQkUGB47Mt1hSI,1343
 ai_edge_torch/generative/examples/phi/verify_util.py,sha256=te2ws_4ZsuMr_d3CkN-vCIAnWGq7g0eDJ0RLna3E2gw,3037
 ai_edge_torch/generative/examples/qwen/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/qwen/convert_to_tflite.py,sha256=XQee8T78I4CqFghZw6sSsrtryWcA_NlOzg2wMfHzHwY,2064
-ai_edge_torch/generative/examples/qwen/convert_v3_to_tflite.py,sha256=P10_5tnuvjXNYyWTp0Fzzj-8moFWyONDBhcGi2yvdp4,2070
+ai_edge_torch/generative/examples/qwen/convert_to_tflite.py,sha256=IM6JgxOuWA5XKMaXKT5j-JK2cm0hrMjrs-hN-fIQXA4,1325
+ai_edge_torch/generative/examples/qwen/convert_v3_to_tflite.py,sha256=7V77Nyd3lldukopxKqru7hD3oH6zyzkRI1fzFl7uZOE,1331
 ai_edge_torch/generative/examples/qwen/qwen.py,sha256=VrrTS83q1Fzp34SjUf0pFiCoMwXehF7VJCZjbxbQ5vA,4332
 ai_edge_torch/generative/examples/qwen/qwen3.py,sha256=cJcN5nH-yAglKtm5Yc5NsLMQvkJ4Drz_IsQJzS91Veo,5329
 ai_edge_torch/generative/examples/qwen/verify_qwen2.py,sha256=ry-c2QesH-0KnrSQygfjUFs6d4kOFvJz2ts_8mP156I,1659
@@ -133,15 +133,15 @@ ai_edge_torch/generative/examples/qwen_vl/verify.py,sha256=m-SdwHlxjOHsw740_i3T6
 ai_edge_torch/generative/examples/qwen_vl/verify_decoder.py,sha256=ZuVRPLnxpaJWAvZLvJa-u1hZIkqc1D6Ew8FxdXtLlyM,2663
 ai_edge_torch/generative/examples/qwen_vl/verify_image_encoder.py,sha256=PZ392nDoJG2OmHZ_7Jet3Zu1JkN6QErxKcDc7a-PPds,3126
 ai_edge_torch/generative/examples/smollm/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/smollm/convert_to_tflite.py,sha256=CsXboOmzk2GB506a-Je-Js-XFlUmikVX7dhBFQAmiUs,2040
-ai_edge_torch/generative/examples/smollm/convert_v2_to_tflite.py,sha256=2BFScI1RbpK3YqJcsvacGz9haKR_gwGsSNHfYmlOvNg,2045
+ai_edge_torch/generative/examples/smollm/convert_to_tflite.py,sha256=QJB0iDBJoxP292O60-E57AbV57YdxwL9K1ZsTzgOUe8,1069
+ai_edge_torch/generative/examples/smollm/convert_v2_to_tflite.py,sha256=b7TjJZI0gN2rPv4pk5Iw3T-vTsc1UAGGRWppOskNl0E,1074
 ai_edge_torch/generative/examples/smollm/smollm.py,sha256=3QbVefwSlFjhze7ju3P6xgMex2d1AJpF6Ii_qlbfJ34,3656
 ai_edge_torch/generative/examples/smollm/verify.py,sha256=tXiAnwOnqgwyoa8dI4tCBiGUXkOMfdE9MUkkY_Bc4Ig,1603
 ai_edge_torch/generative/examples/smollm/verify_util.py,sha256=tHwmo8E474gLKAu4IrG7t2hmy-3Go_BJI1mgu0KkWOY,3076
 ai_edge_torch/generative/examples/stable_diffusion/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/stable_diffusion/attention.py,sha256=kDWG6MlIGa89zC5KSRcJlw2c4ITuw8KcchtfmF55f4g,3545
 ai_edge_torch/generative/examples/stable_diffusion/clip.py,sha256=lSCRZsoLjH_kqasRMwCy5IogkhyJdwcHKsPEfyxsXCQ,6112
-ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py,sha256=_yk6wVoZm1_FRMFJF5URaPZNNdmMR89fwmKz81BEyao,5601
+ai_edge_torch/generative/examples/stable_diffusion/convert_to_tflite.py,sha256=Wba7-rEibvc7ii0uQhO87meVqdQC0_3mtYj5PRXXXbE,5584
 ai_edge_torch/generative/examples/stable_diffusion/decoder.py,sha256=afyHXc86h-ij5zTULmZnM1h313N9VWCyIVriH6pqeSo,16368
 ai_edge_torch/generative/examples/stable_diffusion/diffusion.py,sha256=ylqXOZhYc6XFCaNBKQw0jAnYrCtRFFQKzQzEsFIntvo,34890
 ai_edge_torch/generative/examples/stable_diffusion/encoder.py,sha256=CAPsW84A8f00nS6fLFeh_XUjCPsDCA5UxHOUsMrLfSU,3450
@@ -151,7 +151,7 @@ ai_edge_torch/generative/examples/stable_diffusion/util.py,sha256=XIXIB0vCvQKOGy
 ai_edge_torch/generative/examples/stable_diffusion/samplers/__init__.py,sha256=uQWKzCD_49ackNFrt50H04dkDXxfAwUCtMWWQre5SVE,830
 ai_edge_torch/generative/examples/stable_diffusion/samplers/k_euler.py,sha256=wBBNM24waZ57M1rXonwesfUkKe9DqpqO3eW6BfZkrD0,2323
 ai_edge_torch/generative/examples/stable_diffusion/samplers/k_euler_ancestral.py,sha256=c89ldwtuQ2_yspGrGa7oh7fsvTt6A86Whxa6fBK9YOQ,2526
-ai_edge_torch/generative/examples/stable_diffusion/samplers/k_lms.py,sha256=ZE6HyOoBJrmTh54KVFf7DjNBnBS0pT4cgviYaq8HGMU,2801
+ai_edge_torch/generative/examples/stable_diffusion/samplers/k_lms.py,sha256=NDiJPreRFHW1l50Lmb1FHNQcs3wslFAhS4FEFTQKjMU,2805
 ai_edge_torch/generative/examples/stable_diffusion/samplers/sampler.py,sha256=RxR5rw0wFFm_5CfAY-3-EIz83vhM9EKye8Bb5zBb0Ok,1341
 ai_edge_torch/generative/examples/t5/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/examples/t5/convert_to_tflite.py,sha256=HHtZTtUh3QgE4F74-ru_8n1pt6cqfbObw12xoaMJ7NQ,4596
@@ -162,56 +162,56 @@ ai_edge_torch/generative/examples/test_models/convert_toy_model.py,sha256=6-WaNH
 ai_edge_torch/generative/examples/test_models/toy_model.py,sha256=s3l__g3l0DpgXEPs1ikcJqSS7OfWzgYdkOLnEwdjcUo,5257
 ai_edge_torch/generative/examples/test_models/toy_model_with_kv_cache.py,sha256=rVTaNQxlf3-mv9e6s33V9CDd_cmVwPti1A3ARUAwSD4,4766
 ai_edge_torch/generative/examples/tiny_llama/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/examples/tiny_llama/convert_to_tflite.py,sha256=dYqb9OhpZUBIdww1lUdMjFhEo3IYhpvitvrZfO0uGSs,1827
+ai_edge_torch/generative/examples/tiny_llama/convert_to_tflite.py,sha256=rAVdo2U-4TInZPfZmn3-7rCmo94bY9nPN-85mTqFR_w,1088
 ai_edge_torch/generative/examples/tiny_llama/tiny_llama.py,sha256=fRuPLHwZu8Am8h0Y3dT5_1vsCEnm6euGWB_KHc3xBXk,2735
 ai_edge_torch/generative/examples/tiny_llama/verify.py,sha256=qzUsW8tJlAD9mqRxDSAcz5xSUKNlBz_DykA3PwUHMwc,1336
 ai_edge_torch/generative/examples/tiny_llama/verify_util.py,sha256=z6vPBXDWAL6gN45z-Wm1B2bnZuJf3ErruSuFaEPPEGY,2994
 ai_edge_torch/generative/fx_passes/__init__.py,sha256=PFSMsA1vfBfrV9ssBCkYJNl8Hx_bLdWjN01iyjPM5jE,1094
 ai_edge_torch/generative/fx_passes/remove_sdpa_zero_mask_pass.py,sha256=myGjal5A8yIBoqgArd2k40rZmCgD1Ya369KR7182bhI,2129
 ai_edge_torch/generative/layers/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/layers/attention.py,sha256=RaXENRRQo1MsLdt3U8h3kYTCmd6imHQ-aCXtmPXCh_o,13911
-ai_edge_torch/generative/layers/attention_test.py,sha256=9v8v96TLyFPdqxEylU1JOAeRFAp2s0YoDHZN83SFJJA,4764
-ai_edge_torch/generative/layers/attention_utils.py,sha256=2qfg7Tzk9ikKph5w3geOHC1I6EyOCdDsWXMr7F7IOZM,7630
-ai_edge_torch/generative/layers/attention_utils_test.py,sha256=22gQ1gcRPkwqFG3_p82GZfRKVE3udEssSy58wNOqv0w,2431
-ai_edge_torch/generative/layers/builder.py,sha256=UiLOyvwd-bc0n5XcbVxi6JCn_qiKSC6zrDKSZT_TSDA,5030
+ai_edge_torch/generative/layers/attention.py,sha256=ZjU3vX-7gOq1KQb3xSZ1NT3xryOTXbYb_vkx_DlcizA,14524
+ai_edge_torch/generative/layers/attention_test.py,sha256=ON9jQRY1r2kFpVq-Qkg6b13Ob95fd4PqHo1hic3RbOQ,5057
+ai_edge_torch/generative/layers/attention_utils.py,sha256=3Ox1XjW_vaqz1-RuVG9RbzRKUqCberFW8P2BQcoNm7A,9659
+ai_edge_torch/generative/layers/attention_utils_test.py,sha256=IHIk39wqaPvxmkZtW27VD3_4xUpyFow_7mScf8OWdqU,3292
+ai_edge_torch/generative/layers/builder.py,sha256=5QL59CbOOW_mk3mlPdcdirGcAxdLee5atbZlnu5Z3ts,5079
 ai_edge_torch/generative/layers/einsum.py,sha256=LH4CNHr-pFfLUuCpwbYL3GpoAMgHJ4nLju3XCqA4VwM,1416
 ai_edge_torch/generative/layers/einsum_test.py,sha256=ltIE773bvvNLv_9aLQxFwe1MgQ762sez0c5E2tejxuA,1079
 ai_edge_torch/generative/layers/feed_forward.py,sha256=_GmtHxwL068l9gh_F_WFcFk7La-Tl5SfoQ9v2hMabZM,5541
 ai_edge_torch/generative/layers/feed_forward_test.py,sha256=Y5l1eC9NgfYixHcfIfE1W4FGh7oC-9UGGyHdKS9tQKc,1880
 ai_edge_torch/generative/layers/kv_cache.py,sha256=A0IFXZ1HD2ZHOWRLfsDO4almgE0KQfjyBOdBFZIGnAs,10893
 ai_edge_torch/generative/layers/lora.py,sha256=hsvWLLOnW7HQ0AysOZu30x_cetMquDd1tjfyLz8HCSU,17892
-ai_edge_torch/generative/layers/model_config.py,sha256=HP-vu1UmAiTmdLlTyZGDUF3le0gji8a61mLCy966NZw,10261
-ai_edge_torch/generative/layers/normalization.py,sha256=WAhPcLbcC3SCEa6oIgdsojvN306_S8d90WyMQ7ZVP6I,7269
+ai_edge_torch/generative/layers/model_config.py,sha256=y6nyydMzm7JUV7AKttcFy3tvti-nE6tRXoVbBB9dyiM,10438
+ai_edge_torch/generative/layers/normalization.py,sha256=syasVh3dRDVp2Nwhl0x7zucL-chTnCqWgeV1mb87DFY,7435
 ai_edge_torch/generative/layers/normalization_test.py,sha256=zwurZly-TgFxdgVVdpzu9vCpcLbd5RYt_gKg9Lfg1jI,2248
 ai_edge_torch/generative/layers/rotary_position_embedding.py,sha256=975zR202MdIrILJ7blceAcxrNqX1ZCN0ECKG1gz-bV8,2655
-ai_edge_torch/generative/layers/scaled_dot_product_attention.py,sha256=2_AgwENsaOgaxgiSqgoj0V0JzQ09dFtP_nBhX-lJK2g,5648
+ai_edge_torch/generative/layers/scaled_dot_product_attention.py,sha256=1zhOsJpI4CTn78weOs0uRwkRxYu6wGfBvYVFpGFl0qQ,6681
 ai_edge_torch/generative/layers/scaled_dot_product_attention_test.py,sha256=c6JBMQsq9XeMmR1XvGEIidNsoh-YIvichXo2LwVHgr4,3301
-ai_edge_torch/generative/layers/sdpa_with_kv_update.py,sha256=iw7D_46CFe9iRvU0UumbkIoqWQEhDroxm9ABcK-CLlM,3600
+ai_edge_torch/generative/layers/sdpa_with_kv_update.py,sha256=fK_h9M-03ai5dV8ZyQzvB0y84IKlNg9h-4bt9F6bU0g,3833
 ai_edge_torch/generative/layers/unet/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/layers/unet/blocks_2d.py,sha256=ZteHZXK6HKyxYji49DQ46sA9aIy7U3Jnz0HZp6hfevY,28996
 ai_edge_torch/generative/layers/unet/builder.py,sha256=zAqWXdimmMrQRhmE_t9XkS68mh6PSrzwb-2NZZXrR5I,1901
 ai_edge_torch/generative/layers/unet/model_config.py,sha256=pPDwLawc23pfMaPVyMJlYmxVVusjMvx-l8wBwOYOH-c,9692
 ai_edge_torch/generative/quantize/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/generative/quantize/example.py,sha256=1lfVNUd2cEyRUnoZ7BLbRJ9IN-FTKiWBtZNPFUzAiWE,1747
-ai_edge_torch/generative/quantize/quant_attrs.py,sha256=plMsd7JBi98r2NHsAdMdvS6TPTXAoRFLCwOXu8H3-24,2004
-ai_edge_torch/generative/quantize/quant_recipe.py,sha256=CEW-ewHxwb59x_GISx4jr7WMihvn-jKWVcBonllzDS4,5724
-ai_edge_torch/generative/quantize/quant_recipe_utils.py,sha256=h3k_na6rbR08Ip79-2JbkeH8RDk_rrnEGiytuzFDhqc,2678
-ai_edge_torch/generative/quantize/quant_recipes.py,sha256=mmPsgQ2vloMWflqJ6ALmD1lANacTkWotEhurb5foK30,2771
-ai_edge_torch/generative/quantize/supported_schemes.py,sha256=TwR2FpQuBEORy6FshEyHNBMKARWlA2MVtTfX9tXV5aE,1488
+ai_edge_torch/generative/quantize/example.py,sha256=rRm5noeqnCJyaDYG6e1micwaYdA3ooXK0MBYl98neQQ,1742
+ai_edge_torch/generative/quantize/quant_attrs.py,sha256=THKS82G9sR25biEk8AVRFWwB1cjVXpbjZD5IBfNOOMw,2344
+ai_edge_torch/generative/quantize/quant_recipe.py,sha256=kL3UPNscWKzugiyunMn5XK6Lzpw-yXTdtM_NmCNbFUM,5330
+ai_edge_torch/generative/quantize/quant_recipe_utils.py,sha256=bFvY-OnbH6fe5xhS6dfxkAl7NTGzOVtGJ_8417GwGgI,2478
+ai_edge_torch/generative/quantize/quant_recipes.py,sha256=nvADL7wObxJLp651HI1xv5_HeQZ2RaY2swoRC1EOK7k,2783
+ai_edge_torch/generative/quantize/supported_schemes.py,sha256=lEcqwxOIfxhTYCQUelODLky57933r372Uh7sHYNV0Ok,1712
 ai_edge_torch/generative/test/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/generative/test/test_custom_dus.py,sha256=MjIhTvkTko872M35XMciobvICcDWTcIDJ3rociko-wM,3267
-ai_edge_torch/generative/test/test_kv_cache.py,sha256=zVocQZTORIROwSYVMmy5KGPDl6kTuEGko7L1PTkua8g,5568
+ai_edge_torch/generative/test/test_kv_cache.py,sha256=kdoBpyzVOjQsZkEPnbDOh33gRE0RR8tq1XCgJkSYwU0,6157
 ai_edge_torch/generative/test/test_loader.py,sha256=9mQUeeZKOVApOWSWl2cN9c10axZjMKM1-0Zd823CCS4,3449
 ai_edge_torch/generative/test/test_lora.py,sha256=sKnBixmGIXHUtOwh3SA4MIFenPbjK2n-Xknwic_KMDQ,5046
 ai_edge_torch/generative/test/test_model_conversion.py,sha256=T35zdzag2-nmy4qc6AifAjbDXAHU2vyLTE1QCabYBzk,6298
 ai_edge_torch/generative/test/test_model_conversion_large.py,sha256=NkEwrjO8vIcde3XwanpFBhNIw1GSOyJFKNjlvSJmVMY,13271
-ai_edge_torch/generative/test/test_quantize.py,sha256=kKJ01wscTC2t_Ylr7huO5gNKES01gm3dT1gx52z15PA,7356
+ai_edge_torch/generative/test/test_quantize.py,sha256=Sh82oxDIc_vw-rCd991qMNQxzM3OkI3oiPj2g4hD5MQ,7088
 ai_edge_torch/generative/test/utils.py,sha256=tF6aCfAGJnc9dmzCnZCEOuKNVimfWOqscv9og0DDLHU,2656
 ai_edge_torch/generative/utilities/__init__.py,sha256=-_jxnnFnCgnTU4oTm4MnRsvL5lqhomBNdFBbqfmfHPo,720
-ai_edge_torch/generative/utilities/converter.py,sha256=Q2C9DqYWqiyuu8KoL69SAxDL-RNJ7K8_Z45H2FpsAn0,17563
-ai_edge_torch/generative/utilities/export_config.py,sha256=qjkEbjcvi2AgQikZS5qfgR95Z5z9pm07KX-RN5ibfNE,2280
-ai_edge_torch/generative/utilities/litertlm_builder.py,sha256=0cNuaqhc7cQcAa4NRalUXyoPQUQC9O3-aHAJEDV1Mps,4265
-ai_edge_torch/generative/utilities/loader.py,sha256=drgKBmNibuc3PCdc0kU0pVcp2Nt1_mjLYh67RyXOn7U,15952
+ai_edge_torch/generative/utilities/converter.py,sha256=B2DNg5LcDoniHnh22KFDDSHqnmi8w87-dstKCae3CeM,23322
+ai_edge_torch/generative/utilities/export_config.py,sha256=5B15nYyqf96kjjYlHfPctUfsIdsBsh1f8rxKitJpwKQ,2384
+ai_edge_torch/generative/utilities/litertlm_builder.py,sha256=Jy6R4S18JCIk77ZwrfzeOCSaGozvRQfXSobGkwOEOKs,6447
+ai_edge_torch/generative/utilities/loader.py,sha256=QQeEu0cTC7gWnB7RkHonjWLdVGjMbDHd1lfYO_TcJyU,16047
 ai_edge_torch/generative/utilities/model_builder.py,sha256=xBvcTxihB9TN88UtQiXA9sAITQgf-pA77R-VZlLgUeU,6950
 ai_edge_torch/generative/utilities/moonshine_loader.py,sha256=_RpFabSqtGH5PHiP3_1f6QfO14qMADUxr_HGRlVDFB0,4891
 ai_edge_torch/generative/utilities/stable_diffusion_loader.py,sha256=dqPD9qRXEWtU3ombslOC-BE2l_dMwHoCNu7NsIJhsso,36158
@@ -232,11 +232,11 @@ ai_edge_torch/lowertools/common_utils.py,sha256=4HQtquPZ6oiId8vR_1ykW_uK4ELnyo5z
 ai_edge_torch/lowertools/odml_torch_utils.py,sha256=QRuS7S5lULRWEh3J1sWIsnKh-rbX7rd9tt6JJHbMPfo,8317
 ai_edge_torch/lowertools/test_utils.py,sha256=mdxTlhqHABZEQ_GEmPFCL8LIAWtqRtYZUGdSY1ieZjw,1949
 ai_edge_torch/lowertools/torch_xla_utils.py,sha256=1EytIw2R6dthhLhf69wN1L9BaQTeybCD0wga-PhHcMI,9518
-ai_edge_torch/lowertools/translate_recipe.py,sha256=Oavs0ENKVnIeB-WidXvokTPqNlFfOuP0GMV0d5RK2Rg,6285
+ai_edge_torch/lowertools/translate_recipe.py,sha256=WEv8Kr0RZw0pBc5PoNUEPnQGBW_JUmuVWQ7Hf4bw0uY,6545
 ai_edge_torch/odml_torch/__init__.py,sha256=S8jOzE9nLof-6es3XDiGJRN-9H_XTxsVm9dE7lD3RWo,812
 ai_edge_torch/odml_torch/_torch_future.py,sha256=jSYHf1CMTJzMizPMbu2b39hAt0ZTR6gQLq67GMe9KTo,2336
 ai_edge_torch/odml_torch/_torch_library.py,sha256=Lw1gqL2HWNRspdTwNhIkYAHDyafHedHtkXyKKxn-Wss,805
-ai_edge_torch/odml_torch/export.py,sha256=WPsATMFsc62aETExBsb1v6Ec1fL3r0CEvxwE7r2C_CA,14931
+ai_edge_torch/odml_torch/export.py,sha256=PdNdFtEZcFpy4G8ikhElnil8yoYZUWHQ_180E1BGFV4,15347
 ai_edge_torch/odml_torch/export_utils.py,sha256=Eax4QUefIzpmVuQxo1y9FqJ6g0qXjg4C0IVZ5uYPscs,4899
 ai_edge_torch/odml_torch/optimization_barrier.py,sha256=2lmSiu5iXWLFWpupZHvsVeNYNzG5AVGSK3K_CNhS5Sk,2290
 ai_edge_torch/odml_torch/tf_integration.py,sha256=NN29WeXmHZ0S1RPDFHUnBi2DEjMvAtwczStPYIsQ1w8,4849
@@ -246,15 +246,21 @@ ai_edge_torch/odml_torch/composite/stablehlo_composite_builder.py,sha256=h6DQkYV
 ai_edge_torch/odml_torch/debuginfo/__init__.py,sha256=3A_lMyj-B-DOhLJG6WmjKvZK5te2rXje8FrfqOhZsN0,959
 ai_edge_torch/odml_torch/debuginfo/_build.py,sha256=6Ns2rlfOilLJEk5cUxlkRwm2uxOgEF2-0S2DMcOqr6A,3319
 ai_edge_torch/odml_torch/debuginfo/_op_polyfill.py,sha256=IvOBQyROI9WHS3umHRxsDW-1YElU9BPWzKtJA2eKWOI,1739
+ai_edge_torch/odml_torch/experimental/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
+ai_edge_torch/odml_torch/experimental/torch_tfl/__init__.py,sha256=MN-8wS8vZmSWr9TOoZivqzZGDPoNpy15dByM3arNnM8,968
+ai_edge_torch/odml_torch/experimental/torch_tfl/_decomps.py,sha256=KEPc8wZeAukFhlfKUA2TSmRdMozqG16msnEw7JsGrTg,13256
+ai_edge_torch/odml_torch/experimental/torch_tfl/_lowerings.py,sha256=9coRYoVQFtGhcWuMgZf7lRGIMpRqupbyueAqShx6ieQ,21065
+ai_edge_torch/odml_torch/experimental/torch_tfl/_ops.py,sha256=gRoAf6QjER-fb4oVB05vVGDV__HiXeJgEo1WXX7TnkM,10778
+ai_edge_torch/odml_torch/experimental/torch_tfl/torch_library_utils.py,sha256=9Oq3Dsc5YAkpBgPNFWTquT4TVQnOG8LOlwIabDIc10k,1236
 ai_edge_torch/odml_torch/jax_bridge/__init__.py,sha256=e9Oa4J3An9FYr3zM0OzjzyNNitEeg-IoBUGNfUxsDSA,798
 ai_edge_torch/odml_torch/jax_bridge/_wrap.py,sha256=LqwZ1vCJTSOzgzvH8LUAN-sAkF-l_pGj1AMEIzAqHCA,6638
 ai_edge_torch/odml_torch/jax_bridge/utils.py,sha256=T8isGc896VrHZ6c_L5pYmLpolQ7ibcOlgWfPuVFPzIg,2264
 ai_edge_torch/odml_torch/lowerings/__init__.py,sha256=uJ-niilt1c-D6QJzLwgvCUf62le_JsxQTlqj_iP_Ps0,1009
-ai_edge_torch/odml_torch/lowerings/_basic.py,sha256=sC4N5-7RS9yKecs97kM9J56enGvsZj1CJo7y79cuzRg,12784
+ai_edge_torch/odml_torch/lowerings/_basic.py,sha256=HOTYfQWin8tqi1yakIyardxhRViZ6rhLV6ZomMSS7zA,17554
 ai_edge_torch/odml_torch/lowerings/_batch_norm.py,sha256=PaLI0BB6pdBW1VyfW8VTOT_Be-ZcqYdNOsyfzKfq8Cg,2064
 ai_edge_torch/odml_torch/lowerings/_convolution.py,sha256=Q0aDzyUcZMoSzSbOU-r3LJMgPe6fble0QwdYVIOHHHk,6887
 ai_edge_torch/odml_torch/lowerings/_decomp_registry.py,sha256=ybOdoFE5HIJTkyiYcc73zpyUyUpioVnAca6k0wyJPs4,2572
-ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py,sha256=6KZc0-SM0LupOnDAgnUjj47aHk7YJ0cC1ulAeWvV5_w,19289
+ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py,sha256=YmyM-5HJeeYaIhmKTOnCjfX3_A1PPh1gPGUi1d8EBs8,26454
 ai_edge_torch/odml_torch/lowerings/_layer_norm.py,sha256=khJIvDVk2s332Nd2Be-5dM6-wp5DGff61HCV5lskHmQ,3011
 ai_edge_torch/odml_torch/lowerings/_quantized_decomposed.py,sha256=XDZ0zLej_XaQDJnaAAxhNFAd7NfQm5SOVEp_nno_krA,6178
 ai_edge_torch/odml_torch/lowerings/_rand.py,sha256=g6SuqDkuC6hD35lyP1-5H7ASDIzPSmKukeNT5naZSv8,4133
@@ -270,8 +276,8 @@ ai_edge_torch/testing/__init__.py,sha256=_yGgvnBZWb7T3IN3mc4x1sS4vM96HZwM8pwIcPG
 ai_edge_torch/testing/export.py,sha256=k5mGDGzwc23Z4zaIVDs8CNh-oOt64gsf9MS9NjhbPy4,3293
 ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
 ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
-ai_edge_torch_nightly-0.7.0.dev20250929.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.7.0.dev20250929.dist-info/METADATA,sha256=VztFMGggEpsejPdHnw6bnlXQu8vY6nVWL8oGXbxwgW0,2074
-ai_edge_torch_nightly-0.7.0.dev20250929.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_torch_nightly-0.7.0.dev20250929.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.7.0.dev20250929.dist-info/RECORD,,
+ai_edge_torch_nightly-0.8.0.dev20251206.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.8.0.dev20251206.dist-info/METADATA,sha256=-BM6OyCHwhI74Y8OYtnKNLt6dYiKGkEr1mSZK5hv9Lg,2399
+ai_edge_torch_nightly-0.8.0.dev20251206.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+ai_edge_torch_nightly-0.8.0.dev20251206.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.8.0.dev20251206.dist-info/RECORD,,

{ai_edge_torch_nightly-0.7.0.dev20250929.dist-info → ai_edge_torch_nightly-0.8.0.dev20251206.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any