Commits

History for

Commits on Apr 19, 2021

Move jax.config to jax._src.config.

hawkinsp
authored and
jax authors
committed

Commits on Mar 29, 2021

remove disable_omnistaging mechanism
mattjj
committed

Commits on Jan 4, 2021

[host_callback] Added host_callback.call to return values to the device
gnecula
committed

Commits on Jul 30, 2020

omnistaging, under a flag and disabled by default (jax-ml#3370 )
mattjj
authored

Commits on Jul 14, 2020

Cleanup: convert uses of 'import numpy as onp' in tests (jax-ml#3756 )
jakevdp
authored

Commits on Jul 4, 2020

Refactored host_callback to use the C++ runtime. (jax-ml#3644 )
gnecula
authored

Commits on Jun 29, 2020

Commits on Jun 1, 2020

Make check_dtypes, atol, and rtol keyword-only arguments in jax.test_… (jax-ml#3280 )
hawkinsp
authored

Commits on May 11, 2020

Update JAX to avoid XLA:Python API names deprecated in jaxlib 0.1.46. (jax-ml#3046 )
hawkinsp
authored

Commits on Apr 23, 2020

Remove usage of xla_client.{Computation,ComputationBuilder}. (jax-ml#2808 )
hawkinsp
authored

Commits on Jan 29, 2020

Remove "from __future__" uses from JAX. (jax-ml#2117 )
hawkinsp
authored

Commits on Oct 9, 2019

Add infeed test case.
hawkinsp
committed

Loading