JAX 0.12.0 Adds an IFRT Proxy Backend and Custom VJP Rematerialization, and Requires Haswell-Class x86-64 CPUs and NumPy 2.2
JAX 0.12.0, tagged October 19, adds erfcinv, a polynomial primitive, defremat and a proxy backend, and removes jax.custom_remat and support for AVX-only CPUs.