jax

abc6599e - an experiment in handling instances with __array__