Commits

Branch selector

 main

User selector

hawkinsp

All time

Commit History

Commits on Aug 18, 2024

• [numpy] Fix test failures under NumPy 2.0.

  Show description for ba5b081
  

  hawkinsp

  authored and

  jax authors

  committed
  ba5b081

  Copy full SHA for ba5b081

Commits on Aug 15, 2024

• Reverts 6fc57c0eb6f06b2da20c94f5f127fe4a551bda09

  Show description for 1516d59
  

  hawkinsp

  authored and

  jax authors

  committed
  1516d59

  Copy full SHA for 1516d59

Commits on Aug 14, 2024

• Fix test failures.

  Show description for 323e257
  

  hawkinsp

  authored and

  jax authors

  committed
  323e257

  Copy full SHA for 323e257

Commits on Jul 31, 2024

• Fix or disable some tests that fail when using a Eigen BLAS with AVX vectorization.

  Show description for 858dc54
  

  hawkinsp

  authored and

  jax authors

  committed
  858dc54

  Copy full SHA for 858dc54

Commits on Jul 30, 2024

• Drop support for mhlo in JAX's public API.

  Show description for c1cd7f9
  

  hawkinsp

  authored and

  jax authors

  committed
  c1cd7f9

  Copy full SHA for c1cd7f9

Commits on Jul 29, 2024

• Bump the minimum CUDNN version to v9.1.

  Show description for d1c0d99
  

  hawkinsp

  authored and

  jax authors

  committed
  d1c0d99

  Copy full SHA for d1c0d99

• Bump minimum SciPy version to 1.10.

  Show description for fd23b87
  

  hawkinsp

  authored and

  jax authors

  committed
  fd23b87

  Copy full SHA for fd23b87

Commits on Jul 25, 2024

• Fix "unhashable type" error when passing a jax array as the "repeats" argument to jnp.repeat().

  Show description for 1ac2085
  

  hawkinsp

  authored and

  jax authors

  committed
  1ac2085

  Copy full SHA for 1ac2085

• Simplify jaxpr for jnp.repeat in scalar repeat case.

  Show description for f07e963
  hawkinsp

  committed
  f07e963

  Copy full SHA for f07e963

Commits on Jul 24, 2024

• Simplify promote_shapes.

  Show description for 52fa165
  hawkinsp

  committed
  52fa165

  Copy full SHA for 52fa165

• Don't broadcast scalar conditions in the jnp.where implementation().

  Show description for 7527101
  hawkinsp

  committed
  7527101

  Copy full SHA for 7527101

• Simplify implementation of _broadcast_to.

  Show description for 34ce9f2
  hawkinsp

  committed
  34ce9f2

  Copy full SHA for 34ce9f2

Commits on Jul 23, 2024

• Skip some Pallas tests that fail on TPUv6.

  Show description for fd85c78
  

  hawkinsp

  authored and

  jax authors

  committed
  fd85c78

  Copy full SHA for fd85c78

Commits on Jul 18, 2024

• Don't mask out zero elements on the diagonal of the matrix when inverting triangular matrices.

  Show description for 47e6da3
  

  hawkinsp

  authored and

  jax authors

  committed
  47e6da3

  Copy full SHA for 47e6da3

Commits on Jul 15, 2024

• Disable some tests that fail on Cloud TPU.

  hawkinsp

  committed
  f488c4c

  Copy full SHA for f488c4c

• Bump some SVD test tolerances.

  Show description for 2f45cd7
  

  hawkinsp

  authored and

  jax authors

  committed
  2f45cd7

  Copy full SHA for 2f45cd7

• Update references to jaxlib_nightly_releases (which is a legacy index and no longer updated) to jax_nightly_releases.

  Show description for 3019467
  hawkinsp

  committed
  3019467

  Copy full SHA for 3019467

• Disable Pallas vmap test that is very slow under tsan.

  Show description for a1f6971
  

  hawkinsp

  authored and

  jax authors

  committed
  a1f6971

  Copy full SHA for a1f6971

Commits on Jul 10, 2024

• Create a bazel visibility list for experimental_array_api.

  Show description for e80d143
  

  hawkinsp

  authored and

  jax authors

  committed
  e80d143

  Copy full SHA for e80d143

Commits on Jul 8, 2024

• Deprecate support for custom lowering rules that return tuple-wrapped ir.Values.

  Show description for 262a4f4
  

  hawkinsp

  authored and

  jax authors

  committed
  262a4f4

  Copy full SHA for 262a4f4

• Don't wrap singleton ir.Types during HLO lowering.

  Show description for 3d5784a
  hawkinsp

  committed
  3d5784a

  Copy full SHA for 3d5784a

Commits on Jul 3, 2024

• Fix lint error at head.

  hawkinsp

  committed
  f408d2a

  Copy full SHA for f408d2a

Commits on Jul 2, 2024

• Prefer `__qualname__` as a pjit_p name.

  Show description for f5290dd
  hawkinsp

  committed
  f5290dd

  Copy full SHA for f5290dd

• Use a class with `__slots__` instead of a NamedTuple in JaxprEqn and SourceInfo, which are two tuples we build frequently.

  Show description for 2350a73
  

  hawkinsp

  authored and

  jax authors

  committed
  2350a73

  Copy full SHA for 2350a73

• Small optimization when forming IR constant.

  Show description for 063581a
  

  hawkinsp

  authored and

  jax authors

  committed
  063581a

  Copy full SHA for 063581a

Commits on Jul 1, 2024

• Don't wrap singleton ir.Values with tuples during HLO lowering.

  Show description for 8ab0c07
  hawkinsp

  committed
  8ab0c07

  Copy full SHA for 8ab0c07

• Don't use the trace context in the prune_closed_jaxpr_outputs cache.

  Show description for 55589fb
  

  hawkinsp

  authored and

  jax authors

  committed
  55589fb

  Copy full SHA for 55589fb

Commits on Jun 28, 2024

• Simplify mlir.dense_int_array.

  Show description for ac3cb6f
  

  hawkinsp

  authored and

  jax authors

  committed
  ac3cb6f

  Copy full SHA for ac3cb6f

Commits on Jun 27, 2024

• Add a direct HLO lowering of remat_p that doesn't call eval_jaxpr.

  Show description for 6170369
  

  hawkinsp

  authored and

  jax authors

  committed
  6170369

  Copy full SHA for 6170369

Commits on Jun 26, 2024

• Run `pyupgrade --py310-plus`.

  Show description for 7f4ef63
  hawkinsp

  committed
  7f4ef63

  Copy full SHA for 7f4ef63

• Run `pyupgrade --py310-plus` on .pyi files.

  Show description for e7e1ddc
  hawkinsp

  committed
  e7e1ddc

  Copy full SHA for e7e1ddc

• Update minimum Python version to 3.10.

  hawkinsp

  committed
  945fde4

  Copy full SHA for 945fde4

• Update Github actions builds to use Python 3.10 instead of Python 3.9.

  Show description for f5f9da3
  

  hawkinsp

  authored and

  jax authors

  committed
  f5f9da3

  Copy full SHA for f5f9da3

Commits on Jun 21, 2024

• Update minimum NumPy version to v1.24.

  hawkinsp

  committed
  7f24837

  Copy full SHA for 7f24837

• [JAX] Add caching to pjit._infer_params.

  Show description for 9e30079
  

  hawkinsp

  authored and

  jax authors

  committed
  9e30079

  Copy full SHA for 9e30079