[jax.lax](https://docs.jax.dev/en/latest/jax.lax.html#module-jax.lax) has identified a set of primitives which permits the [jax.numpy](https://docs.jax.dev/en/latest/jax.numpy.html) subset to be implemented on top of immutable tensors. Implementing an equivalent to the `jax.lax` surface provides a foundation to bootstrap the `jax.numpy` surface.
jax.lax has identified a set of primitives which permits the jax.numpy subset to be implemented on top of immutable tensors.
Implementing an equivalent to the
jax.laxsurface provides a foundation to bootstrap thejax.numpysurface.