Repository navigation
-
use_backend("jax")on macOS now defaults togpu = FALSE, sojax-metal
is no longer selected automatically. -
JAX now registers backend tensor S3 methods for
jax.core.Tracerobjects,
improving compatibility for traced computations. -
On Linux,
tensorflow-cpuis no longer pinned to2.18.*when resolving
Python dependencies.