Commits

Branch selector

 main

User selector

gnecula

Datepicker

All time

Commit History

Commits on Feb 15, 2024

• [shape_polyO] Performance improvements for symbolic dimension manipulations (step 2)

  Show description for eb9caf0
  gnecula

  committed
  eb9caf0

  Copy full SHA for eb9caf0

• [shape_poly] Add support for jnp.split

  gnecula

  committed
  18698a1

  Copy full SHA for 18698a1

• Fix JVP rule for convert_element_type to integer types.

  Show description for ddc2487
  gnecula

  committed
  ddc2487

  Copy full SHA for ddc2487

Commits on Feb 14, 2024

• [shape_poly] Improve the symbolic expressions pretty-printer and parser.

  Show description for ed73560
  gnecula

  committed
  ed73560

  Copy full SHA for ed73560

Commits on Feb 13, 2024

• [shape_poly] Improve the handling of equality external constraints.

  Show description for a650f6c
  gnecula

  committed
  a650f6c

  Copy full SHA for a650f6c

Commits on Feb 12, 2024

• [shape_poly] Performance improvements for symbolic dimension manipulations (step 1)

  Show description for 202bcd3
  gnecula

  committed
  202bcd3

  Copy full SHA for 202bcd3

Commits on Feb 9, 2024

• [shape_poly] Add another micro-benchmark

  Show description for 1e76422
  

  gnecula

  authored and

  jax authors

  committed
  1e76422

  Copy full SHA for 1e76422

Commits on Feb 8, 2024

• [shape_poly] Add limited support for equality explicit constraints.

  Show description for 983bb32
  gnecula

  committed
  983bb32

  Copy full SHA for 983bb32

Commits on Feb 7, 2024

• [shape_poly] Change the representation of symbolic dimension expressions

  Show description for 6f05474
  gnecula

  committed
  6f05474

  Copy full SHA for 6f05474

Commits on Feb 5, 2024

• [shape_poly] Add micro-benchmarks for symbolic shape manipulation.

  Show description for 313caba
  

  gnecula

  authored and

  jax authors

  committed
  313caba

  Copy full SHA for 313caba

Commits on Feb 2, 2024

• [export] Set default native serialization version to 9.

  Show description for fdf227e
  

  gnecula

  authored and

  jax authors

  committed
  fdf227e

  Copy full SHA for fdf227e

Commits on Jan 29, 2024

• [shape_poly] Add a decision procedure for inequalities.

  Show description for e20afac
  gnecula

  committed
  e20afac

  Copy full SHA for e20afac

Commits on Jan 24, 2024

• [shape_poly] Add more tests for reasoning about inequalities.

  Show description for 0bd511d
  gnecula

  committed
  0bd511d

  Copy full SHA for 0bd511d

Commits on Jan 23, 2024

• [shape_poly] Forgot to update CHANGELOG for #19235.

  gnecula

  committed
  af30668

  Copy full SHA for af30668

• [shape_poly] Add support for symbolic constraints on dimension variables

  Show description for 24201ef
  gnecula

  committed
  24201ef

  Copy full SHA for 24201ef

Commits on Jan 15, 2024

• [shape_poly] Improve core.max_dim and core.min_dim

  Show description for a1286d0
  gnecula

  committed
  a1286d0

  Copy full SHA for a1286d0

Commits on Jan 13, 2024

• [shape_poly] Protect shape_poly: rename to _shape_poly.py.

  Show description for 0967a79
  

  gnecula

  authored and

  jax authors

  committed
  0967a79

  Copy full SHA for 0967a79

Commits on Jan 12, 2024

• [shape_poly] Improve and rename export.args_specs.

  Show description for 3b7917a
  gnecula

  committed
  3b7917a

  Copy full SHA for 3b7917a

• [shape_poly] Deprecate shape_poly.PolySpec.

  Show description for 6cac99e
  gnecula

  committed
  6cac99e

  Copy full SHA for 6cac99e

Commits on Jan 10, 2024

• [shape_poly] Introduce is_symbolic_dim and deprecate is_poly_dim.

  Show description for df280a1
  gnecula

  committed
  df280a1

  Copy full SHA for df280a1

Commits on Jan 9, 2024

• [shape_poly] Improve the lexicographic ordering of symbolic expressions

  Show description for b7f82e8
  gnecula

  committed
  b7f82e8

  Copy full SHA for b7f82e8

Commits on Jan 8, 2024

• [shape_poly] Replace non_negative_dim with max_dim and min_dim.

  Show description for 6b7b3a3
  gnecula

  committed
  6b7b3a3

  Copy full SHA for 6b7b3a3

• Fix unused import error, breaks ruff.

  gnecula

  committed
  02acacd

  Copy full SHA for 02acacd

• [export] Refactor the imports for the public API of jax.experimental.export

  Show description for 69788d1
  

  gnecula

  authored and

  jax authors

  committed
  69788d1

  Copy full SHA for 69788d1

• Move export backwards compatibility tests out of jax2tf. Step 3.

  Show description for ed2a839
  

  gnecula

  authored and

  jax authors

  committed
  ed2a839

  Copy full SHA for ed2a839

• [shape_poly] Minor improvements for handling of dilation in convolutions.

  Show description for 45fa502
  gnecula

  committed
  45fa502

  Copy full SHA for 45fa502

• [shape_poly] Improved the tests for inequality comparisons.

  Show description for 3195a06
  gnecula

  committed
  3195a06

  Copy full SHA for 3195a06

• [shape_poly] Improve error messages for shape polymorphism

  Show description for 2b4177d
  gnecula

  committed
  2b4177d

  Copy full SHA for 2b4177d

Commits on Jan 7, 2024

• [shape_poly] Simplify and speed-up the __eq__ functions for symbolic expressions

  Show description for cd0e10f
  gnecula

  committed
  cd0e10f

  Copy full SHA for cd0e10f

• Improve some deprecation error messages

  gnecula

  committed
  cea77f5

  Copy full SHA for cea77f5

• [shape_poly] Cache the hash values for symbolic expressions

  Show description for 2ca7b31
  gnecula

  committed
  2ca7b31

  Copy full SHA for 2ca7b31

• [shape_poly] Fixes for the lexicographic ordering of monomials.

  Show description for f1c87e0
  gnecula

  committed
  f1c87e0

  Copy full SHA for f1c87e0

Commits on Dec 19, 2023

• [shape_poly] Update the jax.ops.segment{max|...} to with with shape polymorphism

  Show description for 30bc5a2
  gnecula

  committed
  30bc5a2

  Copy full SHA for 30bc5a2

• Improve support for JAX_DUMP_IR_TO.

  Show description for bb84e6c
  

  gnecula

  authored and

  jax authors

  committed
  bb84e6c

  Copy full SHA for bb84e6c

Commits on Dec 18, 2023

• Move export backwards compatibility tests out of jax2tf. Step 2.

  Show description for cc2a3eb
  

  gnecula

  authored and

  jax authors

  committed
  cc2a3eb

  Copy full SHA for cc2a3eb

Pagination

Previous

Next