Commits

Branch selector

 main

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Oct 27, 2022

• Increase the minimum jaxlib version to 0.3.22.

  Show description for 320d531
  hawkinsp

  committed
  320d531

  Copy full SHA for 320d531

Commits on Oct 26, 2022

• [JAX] Change the default pmap() ordering to match the ordering of jax.devices() for single-process TPU jobs.

  Show description for bf21391
  

  hawkinsp

  authored and

  jax authors

  committed
  bf21391

  Copy full SHA for bf21391

• Fix FP8 compilation failure in jaxlib stemming from the CUDA/ROCM merge.

  Show description for 0814770
  

  hawkinsp

  authored and

  jax authors

  committed
  0814770

  Copy full SHA for 0814770

• Clarify in JAX Basics that JAX array creation is also an operation that requires accelerator dispatch and converting to a regular Python type is a blocking operation.

  hawkinsp

  committed
  71a384d

  Copy full SHA for 71a384d

• [JAX:CPU] Enable buffer donation on CPU.

  Show description for ce9e009
  

  hawkinsp

  authored and

  jax authors

  committed
  ce9e009

  Copy full SHA for ce9e009

Commits on Oct 25, 2022

• Merge CUDA and ROCM kernel code in jaxlib.

  Show description for a852710
  

  hawkinsp

  authored and

  jax authors

  committed
  a852710

  Copy full SHA for a852710

Commits on Oct 24, 2022

• Move jaxlib cpu kernels under jaxlib/cpu/.

  Show description for 894093c
  

  hawkinsp

  authored and

  jax authors

  committed
  894093c

  Copy full SHA for 894093c

Commits on Oct 19, 2022

• [JAX] Redisable int8 convolution tests on GPU due to CI failures.

  Show description for 02dc25f
  

  hawkinsp

  authored and

  jax authors

  committed
  02dc25f

  Copy full SHA for 02dc25f

• Enable more GPU and TPU tests that pass at head.

  Show description for 8072699
  

  hawkinsp

  authored and

  jax authors

  committed
  8072699

  Copy full SHA for 8072699

Commits on Oct 18, 2022

• Remove JAX custom call implementation of batched triangular solve.

  Show description for 5617a02
  

  hawkinsp

  authored and

  jax authors

  committed
  5617a02

  Copy full SHA for 5617a02

Commits on Oct 14, 2022

• Include column information in Python locations under Python 3.11.

  Show description for ec5bec6
  hawkinsp

  committed
  ec5bec6

  Copy full SHA for ec5bec6

• Add Python 3.11 as a compatible Python version.

  hawkinsp

  committed
  fb72c38

  Copy full SHA for fb72c38

• Drop absl-py as a jaxlib dependency.

  Show description for 4988b31
  hawkinsp

  committed
  4988b31

  Copy full SHA for 4988b31

• Fix lax_autodiff_test to avoid scatters into overlapping ranges.

  Show description for 5d07cbe
  

  hawkinsp

  authored and

  jax authors

  committed
  5d07cbe

  Copy full SHA for 5d07cbe

Commits on Oct 13, 2022

• Set --tsl_link_protobuf=true in .bazelrc to unbreak jaxlib build.

  hawkinsp

  committed
  aa6445e

  Copy full SHA for aa6445e

Commits on Oct 12, 2022

• Make the etils[epath] dependency of JAX optional.

  Show description for 91ea57d
  hawkinsp

  committed
  91ea57d

  Copy full SHA for 91ea57d

• Avoid loading scipy eagerly.

  Show description for 9ab8807
  hawkinsp

  committed
  9ab8807

  Copy full SHA for 9ab8807

• Migrate remaining tests from jtu.cases_from_list to jtu.sample_product.

  Show description for 72f4f38
  hawkinsp

  committed
  72f4f38

  Copy full SHA for 72f4f38

• Reenable some tests disabled in the past because of an LLVM bug.

  Show description for 9bb2c99
  

  hawkinsp

  authored and

  jax authors

  committed
  9bb2c99

  Copy full SHA for 9bb2c99

Commits on Oct 11, 2022

• Port more tests from jtu.cases_from_list to jtu.sample_product.

  hawkinsp

  committed
  0d3277b

  Copy full SHA for 0d3277b

Commits on Oct 10, 2022

• Fix compilation failure in lapack kernel under msan.

  Show description for a3a2206
  

  hawkinsp

  authored and

  jax authors

  committed
  a3a2206

  Copy full SHA for a3a2206

• Add input-output aliasing annotations for LAPACK calls on CPU.

  Show description for 2246887
  

  hawkinsp

  authored and

  jax authors

  committed
  2246887

  Copy full SHA for 2246887

• Add changes accidentally omitted from

  Show description for 2ba0396
  hawkinsp

  committed
  2ba0396

  Copy full SHA for 2ba0396

• Copybara import of the project:

  Show description for c657449
  

  hawkinsp

  authored and

  jax authors

  committed
  c657449

  Copy full SHA for c657449

• Reapply: Use input-output aliasing for jaxlib GPU custom calls.

  Show description for 22cd505
  

  hawkinsp

  authored and

  jax authors

  committed
  22cd505

  Copy full SHA for 22cd505

Commits on Oct 7, 2022

• Revert: Use input-output aliasing for jaxlib GPU custom calls.

  Show description for 2693afa
  

  hawkinsp

  authored and

  jax authors

  committed
  2693afa

  Copy full SHA for 2693afa

• Use input-output aliasing for jaxlib GPU custom calls.

  Show description for 93b839a
  

  hawkinsp

  authored and

  jax authors

  committed
  93b839a

  Copy full SHA for 93b839a

Commits on Oct 6, 2022

• Switch lax_numpy_indexing_test to use jtu.sample_product.

  hawkinsp

  committed
  8107e36

  Copy full SHA for 8107e36

Commits on Oct 5, 2022

• Migrate api_test, lax_numpy_test, and lax_vmap_test to

  Show description for 2c946b3
  hawkinsp

  committed
  2c946b3

  Copy full SHA for 2c946b3

Commits on Oct 4, 2022

• Add an internal jtu.sample_product test decorator.

  Show description for c7e5d3d
  hawkinsp

  committed
  c7e5d3d

  Copy full SHA for c7e5d3d

Commits on Sep 28, 2022

• Update version numbers after release.

  hawkinsp

  committed
  b49e31a

  Copy full SHA for b49e31a

• Disable multiprocess_gpu_test that fails on ROCm.

  hawkinsp

  committed
  f7bafb3

  Copy full SHA for f7bafb3

• jax/jaxlib 0.3.20 release candidate.

  hawkinsp

  committed
  8d86436

  Copy full SHA for 8d86436

• Fix test failure in GPU CI if NCCL_DEBUG is enabled.

  Show description for eabb91e
  hawkinsp

  committed
  eabb91e

  Copy full SHA for eabb91e

Commits on Sep 26, 2022

• Change jax_jit_test to be a jax_test() under Bazel that works across backends.

  Show description for d63a944
  

  hawkinsp

  authored and

  jax authors

  committed
  d63a944

  Copy full SHA for d63a944

Pagination

Previous

Next