Commits

Branch selector

 master

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on May 11, 2021

• [XLA] Fix incomplete gamma functions where x is infinity.

  Show description for 4f9eb64
  

  hawkinsp

  authored and

  jax authors

  committed
  4f9eb64

  Copy full SHA for 4f9eb64

• Promote the x.at[idx].set(y) operators as the preferred way to do indexed updates.

  Show description for d005e38
  hawkinsp

  committed
  d005e38

  Copy full SHA for d005e38

Commits on May 10, 2021

• Pin flatbuffers 1.12 for CI tests.

  hawkinsp

  committed
  40c5e37

  Copy full SHA for 40c5e37

• Allow non-inexact dtypes for jnp.nan...() reductions.

  Show description for d96bf0b
  hawkinsp

  committed
  d96bf0b

  Copy full SHA for d96bf0b

• Improve type error when an object dtype is passed to an operator like +.

  Show description for e6fb6e0
  hawkinsp

  committed
  e6fb6e0

  Copy full SHA for e6fb6e0

• Add support for arm64 builds to build.py and build_wheel scripts.

  hawkinsp

  committed
  7d4551d

  Copy full SHA for 7d4551d

Commits on May 7, 2021

• Allow manylinux_2_12_x86_64 in auditwheel test. This is an alias by PEP600.

  hawkinsp

  committed
  5b3bdd7

  Copy full SHA for 5b3bdd7

• Disable pjit infeed test on CPU and GPU.

  Show description for 195cbeb
  

  hawkinsp

  authored and

  jax authors

  committed
  195cbeb

  Copy full SHA for 195cbeb

Commits on May 6, 2021

• Add support for select and select_range to jax.scipy.linalg.eigh_tridiagonal().

  Show description for bf6b59e
  hawkinsp

  committed
  bf6b59e

  Copy full SHA for bf6b59e

Commits on May 5, 2021

• Add a tridiagonal eigh solver.

  hawkinsp

  committed
  97e89bd

  Copy full SHA for 97e89bd

Commits on May 3, 2021

• Update XLA.

  hawkinsp

  committed
  efe4a1f

  Copy full SHA for efe4a1f

• [JAX] Include pre-transformed stack traces as additional context to JAX exceptions, where present.

  Show description for 2c92bc9
  

  hawkinsp

  authored and

  jax authors

  committed
  2c92bc9

  Copy full SHA for 2c92bc9

• Readd CUDA 10.1 to the set of released CUDA wheels.

  Show description for f9899df
  hawkinsp

  committed
  f9899df

  Copy full SHA for f9899df

Commits on May 1, 2021

• Fix stack trace filtering for paths that do not exist.

  Show description for 89a6b6d
  

  hawkinsp

  authored and

  jax authors

  committed
  89a6b6d

  Copy full SHA for 89a6b6d

• Fix cloudpickle breakage.

  Show description for a80aab0
  

  hawkinsp

  authored and

  jax authors

  committed
  a80aab0

  Copy full SHA for a80aab0

• Update XLA.

  hawkinsp

  committed
  cacc5af

  Copy full SHA for cacc5af

• [JAX] Switch the order of the filtered and unfiltered stack traces in exceptions.

  Show description for e8c3406
  

  hawkinsp

  authored and

  jax authors

  committed
  e8c3406

  Copy full SHA for e8c3406

Commits on Apr 29, 2021

• [JAX] Enable filtered tracebacks on Python 3.6.

  Show description for ccc423d
  

  hawkinsp

  authored and

  jax authors

  committed
  ccc423d

  Copy full SHA for ccc423d

• Bundle libdevice.10.bc with jaxlib wheels.

  Show description for c983d3c
  hawkinsp

  committed
  c983d3c

  Copy full SHA for c983d3c

Commits on Apr 28, 2021

• Fix cusparse kernel build.

  Show description for a0c96b5
  hawkinsp

  committed
  a0c96b5

  Copy full SHA for a0c96b5

Commits on Apr 26, 2021

• Don't build CUDA 11.2 wheels.

  Show description for 79a7f7b
  hawkinsp

  committed
  79a7f7b

  Copy full SHA for 79a7f7b

Commits on Apr 23, 2021

• Disable gmres test on CPU.

  Show description for fb74246
  hawkinsp

  committed
  fb74246

  Copy full SHA for fb74246

Commits on Apr 21, 2021

• Document that JAX follows the NEP-29 deprecation policy.

  Show description for aafe887
  hawkinsp

  committed
  aafe887

  Copy full SHA for aafe887

• Update XLA.

  hawkinsp

  committed
  8654a98

  Copy full SHA for 8654a98

• Handle context manager configuration settings for matmul precision and numpy rank promotion correctly in JIT and linear_util caches.

  Show description for 5261b77
  

  hawkinsp

  authored and

  jax authors

  committed
  5261b77

  Copy full SHA for 5261b77

Commits on Apr 20, 2021

• Remove workaround for NumPy < 1.14.

  Show description for 4409e03
  hawkinsp

  committed
  4409e03

  Copy full SHA for 4409e03

• Disable a test that fails with Numpy 1.17.5, but not Numpy 1.18.0.

  hawkinsp

  committed
  122b2b2

  Copy full SHA for 122b2b2

• Move contents of jax.custom_derivatives to jax._src.custom_derivatives.

  Show description for da1b819
  

  hawkinsp

  authored and

  jax authors

  committed
  da1b819

  Copy full SHA for da1b819

Commits on Apr 19, 2021

• Move jax.config to jax._src.config.

  Show description for 14d991d
  

  hawkinsp

  authored and

  jax authors

  committed
  14d991d

  Copy full SHA for 14d991d

Commits on Apr 15, 2021

• [JAX] Add support for sharing an LRU cache between multiple C++ jit-ted functions.

  Show description for 8a221d3
  

  hawkinsp

  authored and

  jax authors

  committed
  8a221d3

  Copy full SHA for 8a221d3

Commits on Apr 13, 2021

• Update XLA version.

  hawkinsp

  committed
  b37158f

  Copy full SHA for b37158f

• Change scatter shape test to use eval_shape rather than instantiating concrete arrays.

  Show description for 2a3da09
  hawkinsp

  committed
  2a3da09

  Copy full SHA for 2a3da09

• Move jax.api to jax._src.api.

  Show description for 26e9ebc
  

  hawkinsp

  authored and

  jax authors

  committed
  26e9ebc

  Copy full SHA for 26e9ebc

• Enable variadic select_and_gather on TPU.

  hawkinsp

  committed
  0f1520b

  Copy full SHA for 0f1520b

Commits on Apr 12, 2021

• Ensure that profiler tests shut down the profiler, and disable a profiler test.

  Show description for c8184a8
  hawkinsp

  committed
  c8184a8

  Copy full SHA for c8184a8

Pagination

Previous

Next