jax

dd375cbd - Deprecate parsing of __jax_array__ during abstractification.

Deprecate parsing of __jax_array__ during abstractification. Going forward, objects defining __jax_array__ should define pytree lowering if they want to be compatible with JAX transformations.