jax

6a72c522 - Improve docs for jax.numpy: conjugate, conj, imag and real