Commits

Branch selector

 main

User selector

gnecula

Datepicker

All time

Commit History

Commits on Jun 27, 2023

• [shape_poly] Shape polymorphism support for approx_top_k

  Show description for cb42fae
  

  gnecula

  authored and

  jax authors

  committed
  cb42fae

Commits on Jun 26, 2023

• [shape_poly] linalg.schur: shape polymorphism with native lowering on CPU

  Show description for c6a6005
  

  gnecula

  authored and

  jax authors

  committed
  c6a6005

• [shape_poly] linalg.triangular_solve: shape polymorphism with native lowering on CPU

  Show description for a91412e
  

  gnecula

  authored and

  jax authors

  committed
  a91412e

• [shape_poly] Refactor support for dynamic shapes for linalg.eig and linalg.eigh

  Show description for ea0e50f
  

  gnecula

  authored and

  jax authors

  committed
  ea0e50f

• [shape_poly] Refactor some older support for shape polymorphism for linalg.

  Show description for aadcec2
  

  gnecula

  authored and

  jax authors

  committed
  aadcec2

Commits on Jun 25, 2023

• [jax2tf] Improve custom call safety check to detect mhlo.custom_call

  gnecula

  committed
  c800ef8

• [shape_poly] Cleanup the evaluation of dynamic shapes

  Show description for 2299f05
  gnecula

  committed
  2299f05

Commits on Jun 23, 2023

• [shape_poly] linalg.lu: for shape polymorphism for native serialization on CPU.

  Show description for bbc6f30
  

  gnecula

  authored and

  jax authors

  committed
  bbc6f30

Commits on Jun 22, 2023

• [jax2tf] Refactor the backwards compatibility tests.

  Show description for c5a47d1
  

  gnecula

  authored and

  jax authors

  committed
  c5a47d1

• [shape_poly] linalg.cholesky: add support for shape polymorphism with native lowering on CPU

  Show description for f02d122
  

  gnecula

  authored and

  jax authors

  committed
  f02d122

• [shape_poly] linalg.qr: shape polymorphism with native lowering on CPU

  Show description for d940db6
  

  gnecula

  authored and

  jax authors

  committed
  d940db6

• [shape_poly] linalg.svd: shape polymorphism for native serialization on CPU

  Show description for 92288d3
  

  gnecula

  authored and

  jax authors

  committed
  92288d3

• Documentation improvements for XlaCallModule.disabled_checks.

  Show description for 1535fa0
  

  gnecula

  authored and

  jax authors

  committed
  1535fa0

Commits on Jun 20, 2023

• [jax2tf] Rename some backwards compatibility tests.

  Show description for 7a46383
  gnecula

  committed
  7a46383

• Remove unused native_serialization_strict_checks option.

  Show description for 000e6b8
  

  gnecula

  authored and

  jax authors

  committed
  000e6b8

Commits on Jun 19, 2023

• [jax2tf] Refactor the backward compatibility tests.

  Show description for 185abe0
  gnecula

  committed
  185abe0

• [shape_poly] DynamicRngBitGenerator: add shape poly support with native serialization.

  Show description for 5c31d3b
  

  gnecula

  authored and

  jax authors

  committed
  5c31d3b

Commits on Jun 17, 2023

• [shape_poly] DynamicReduceWindow: add shape poly support with native serialization.

  Show description for b83e6fb
  

  gnecula

  authored and

  jax authors

  committed
  b83e6fb

• Add support for dynamic shapes to jax.fft.

  Show description for b9c0658
  

  gnecula

  authored and

  jax authors

  committed
  b9c0658

• [shape_poly] linalg.eig: shape polymorphism with native serialization on CPU

  Show description for 3adfe32
  

  gnecula

  authored and

  jax authors

  committed
  3adfe32

Commits on Jun 16, 2023

• [shape_poly] Refactor and simplify tests.

  Show description for 8435c29
  gnecula

  committed
  8435c29

• [shape_poly] Enable the test for shape polymorphism for custom_linear_solve on TPU.

  Show description for d16b858
  

  gnecula

  authored and

  jax authors

  committed
  d16b858

• [shape_poly] Fix the lowering of lax.dynamic_shapes with shape poly

  Show description for 645b3c4
  gnecula

  committed
  645b3c4

Commits on Jun 15, 2023

• [shape_poly] Change lowering for shape polymorphism to simplify shape refinement

  Show description for 6e09d4a
  gnecula

  committed
  6e09d4a

Commits on Jun 14, 2023

• [shape_poly] Add shape polymorphism support for LU on TPU

  Show description for b147e75
  

  gnecula

  authored and

  jax authors

  committed
  b147e75

Commits on Jun 13, 2023

• [shape_poly] Improve compile-time shape checking.

  Show description for 5b035d5
  gnecula

  committed
  5b035d5

• [jax2tf] Add native_lowering_disabled_checks parameter to jax2tf.convert.

  Show description for 0961fb9
  gnecula

  committed
  0961fb9

Commits on Jun 12, 2023

• [jax2tf] Remove the deprecated experimental_native_lowering option.

  Show description for 9b10384
  

  gnecula

  authored and

  jax authors

  committed
  9b10384

Commits on Jun 6, 2023

• [shape_poly] Enable calling from JAX modules that use shape polymorphism.

  Show description for c46fa27
  

  gnecula

  authored and

  jax authors

  committed
  c46fa27

Commits on Jun 5, 2023

• [shape_poly] Improve DimExpr comparison for "a - mod(b, a) >= 0"

  Show description for d7c0876
  gnecula

  committed
  d7c0876

Commits on Jun 3, 2023

• [shape_poly] Add a polymorphic shape refinement MLIR pass accessible to JAX Python.

  Show description for ec8b855
  

  gnecula

  authored and

  jax authors

  committed
  ec8b855

Commits on Jun 1, 2023

• [jax2tf] Add a backward compatibility test for tf.call_tf_function

  Show description for 37e254e
  

  gnecula

  authored and

  jax authors

  committed
  37e254e

Commits on May 31, 2023

• [shape_poly] Keep track of whether a lowering contains shape polymorphism

  Show description for 5cbc38d
  gnecula

  committed
  5cbc38d

• [shape_poly] Add static constraint checking to the computation of dim vars

  Show description for 9ad8c3b
  gnecula

  committed
  9ad8c3b

Commits on May 26, 2023

• [shape_poly] Add partial support for call_exported with polymorphic shapes

  Show description for 46a258b
  gnecula

  committed
  46a258b

Pagination

Previous

Next