Commits

Branch selector

 main

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Aug 30, 2024

• Disable mosaic gpu tests that are failing at head.

  Show description for cd20404
  

  hawkinsp

  authored and

  jax authors

  committed
  cd20404

  Copy full SHA for cd20404

• Add some msan suppressions to the LAPACK symmetric eigendecomposition FFI call.

  Show description for 1ab3119
  

  hawkinsp

  authored and

  jax authors

  committed
  1ab3119

  Copy full SHA for 1ab3119

• Disable many eigh tests.

  Show description for db4be03
  

  hawkinsp

  authored and

  jax authors

  committed
  db4be03

  Copy full SHA for db4be03

• Relax test tolerance to fix a test failure on ARM CPU.

  Show description for 283770a
  

  hawkinsp

  authored and

  jax authors

  committed
  283770a

  Copy full SHA for 283770a

Commits on Aug 28, 2024

• [pallas] Disable two very slow tests in pallas_vmap_test on CPU.

  Show description for 82c3057
  

  hawkinsp

  authored and

  jax authors

  committed
  82c3057

  Copy full SHA for 82c3057

Commits on Aug 27, 2024

• Fix a number of minor problems in the ROCM build.

  Show description for 45b8719
  

  hawkinsp

  authored and

  jax authors

  committed
  45b8719

  Copy full SHA for 45b8719

Commits on Aug 26, 2024

• Clean up BUILD files.

  Show description for 6d1f51e
  

  hawkinsp

  authored and

  jax authors

  committed
  6d1f51e

  Copy full SHA for 6d1f51e

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 6fc57c0

  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

Pagination

Previous

Next