Commits

Branch selector

 master

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Apr 1, 2021

• Add an api_util hook facility, and add a hook to jax2tf.convert.

  Show description for 94e2c99
  

  hawkinsp

  authored and

  jax authors

  committed
  94e2c99

  Copy full SHA for 94e2c99

Commits on Mar 31, 2021

• Make JAX python code accept C++ DeviceArrayBase classes that don't support weakrefs.

  Show description for 47ff303
  hawkinsp

  committed
  47ff303

  Copy full SHA for 47ff303

Commits on Mar 30, 2021

• Add a JVP rule for the general case of lax.reduce.

  hawkinsp

  committed
  3fc1fdb

  Copy full SHA for 3fc1fdb

• Fix declaration order problem when using JAX_DEBUG_NANS environment variable.

  hawkinsp

  committed
  0cd74c9

  Copy full SHA for 0cd74c9

Commits on Mar 26, 2021

• [JAX] Remove function wrappers in C++ JIT dispatch path.

  Show description for a136f61
  

  hawkinsp

  authored and

  jax authors

  committed
  a136f61

  Copy full SHA for a136f61

Commits on Mar 25, 2021

• Move api_boundary annotation onto C++ jit cache_miss function.

  Show description for f6f0128
  hawkinsp

  committed
  f6f0128

  Copy full SHA for f6f0128

• [JAX] Refactor NaN/Inf checking in jitted functions.

  Show description for cac1b89
  

  hawkinsp

  authored and

  jax authors

  committed
  cac1b89

  Copy full SHA for cac1b89

Commits on Mar 24, 2021

• Switch disable_jit() to use the common boolean state mechanism.

  hawkinsp

  committed
  b1e1d0a

  Copy full SHA for b1e1d0a

• Fix flake8 error.

  hawkinsp

  committed
  99daf13

  Copy full SHA for 99daf13

• Move jax.tree_util implementation to jax._src.tree_util.

  Show description for 6ee6c59
  

  hawkinsp

  authored and

  jax authors

  committed
  6ee6c59

  Copy full SHA for 6ee6c59

• Change enable_x64 to use the common boolean configuration mechanism.

  Show description for 1d2c7b8
  hawkinsp

  committed
  1d2c7b8

  Copy full SHA for 1d2c7b8

• Remove deprecated compatibility code for jaxlib < 0.1.64.

  hawkinsp

  committed
  90e28a1

  Copy full SHA for 90e28a1

• Increase minimum jaxlib version to 0.1.64.

  hawkinsp

  committed
  7052a87

  Copy full SHA for 7052a87

• Fix flake8 error at head.

  hawkinsp

  committed
  166fdd3

  Copy full SHA for 166fdd3

• Remove compatibility code for jaxlib < 0.1.63.

  hawkinsp

  committed
  3c716a6

  Copy full SHA for 3c716a6

Commits on Mar 23, 2021

• Rollforward of:

  Show description for 368f3f0
  

  hawkinsp

  authored and

  jax authors

  committed
  368f3f0

  Copy full SHA for 368f3f0

• Rollback of:

  Show description for 7890d6c
  

  hawkinsp

  authored and

  jax authors

  committed
  7890d6c

  Copy full SHA for 7890d6c

• [JAX] Add an opaque extra_jit_context field to the JAX C++ jit code.

  Show description for f2a6d46
  

  hawkinsp

  authored and

  jax authors

  committed
  f2a6d46

  Copy full SHA for f2a6d46

Commits on Mar 22, 2021

• [JAX] Include jax_enable_x64 in the C++ JIT call signature.

  Show description for e1646bf
  

  hawkinsp

  authored and

  jax authors

  committed
  e1646bf

  Copy full SHA for e1646bf

Commits on Mar 17, 2021

• [JAX] Refactor handling of JIT interpreter state in jax_jit API.

  Show description for 23756a0
  

  hawkinsp

  authored and

  jax authors

  committed
  23756a0

  Copy full SHA for 23756a0

Commits on Mar 16, 2021

• Increase minimum jaxlib version to 0.1.62.

  hawkinsp

  committed
  328930b

  Copy full SHA for 328930b

• Fix test failure with NumPy 1.20.

  Show description for 48a7b15
  hawkinsp

  committed
  48a7b15

  Copy full SHA for 48a7b15

• Add helpful message when import jaxlib fails.

  hawkinsp

  committed
  ee53eeb

  Copy full SHA for ee53eeb

Commits on Mar 15, 2021

• [JAX] Add a .weak_type attribute to C++ array objects.

  Show description for 63c06ef
  

  hawkinsp

  authored and

  jax authors

  committed
  63c06ef

  Copy full SHA for 63c06ef

• [JAX] Enable C++ device arrays by default.

  Show description for 9a2a1ad
  

  hawkinsp

  authored and

  jax authors

  committed
  9a2a1ad

  Copy full SHA for 9a2a1ad

Commits on Mar 12, 2021

• Update PocketFFT version to fix crash due to undersized aligned allocations.

  hawkinsp

  committed
  1ed321c

  Copy full SHA for 1ed321c

Commits on Mar 10, 2021

• Add workaround for SelectAndScatter padding bug on CPU and GPU.

  hawkinsp

  committed
  62a726d

  Copy full SHA for 62a726d

• Readd omnistaging friendly versions of large constant tests.

  hawkinsp

  committed
  5d6ff8b

  Copy full SHA for 5d6ff8b

• Fix crash returning a Token from a jit computation on GPU.

  Show description for 0034939
  hawkinsp

  committed
  0034939

  Copy full SHA for 0034939

• Remove the JAX lazy sublanguage.

  Show description for 140c0ac
  hawkinsp

  committed
  140c0ac

  Copy full SHA for 140c0ac

Commits on Mar 7, 2021

• [XLA:Python] Split DevicePut out of jax_jit and refactor it.

  Show description for 2722837
  

  hawkinsp

  authored and

  jax authors

  committed
  2722837

  Copy full SHA for 2722837

• Cleanups for laziness. No functional changes intended.

  Show description for 2469ad1
  hawkinsp

  committed
  2469ad1

  Copy full SHA for 2469ad1

Commits on Mar 6, 2021

• Delete jax.interpreters.xla.DeviceConstant.

  Show description for 12c2d0d
  

  hawkinsp

  authored and

  jax authors

  committed
  12c2d0d

  Copy full SHA for 12c2d0d

Commits on Mar 4, 2021

• Try to avoid transposes in jnp.einsum by considering both argument orders to dot_general.

  hawkinsp

  committed
  9832df8

  Copy full SHA for 9832df8

• Remove device constants from lazy language.

  Show description for afd2aa2
  hawkinsp

  committed
  afd2aa2

  Copy full SHA for afd2aa2

Pagination

Previous

Next