jax

a3d5fe3c - Delete deprecated jax.lax.lax* modules which were previously private implementation details of jax.