jax
1ba13e1b - Consistently return JAX arrays instead of Numpy-classic arrays from jax.numpy.

Commit
6 years ago
Consistently return JAX arrays instead of Numpy-classic arrays from jax.numpy. Avoids surprising behavior that sometimes arises when mixing the two.
Author
Parents
Loading