Closed jeremiecoullon closed 3 years ago
I assume you meant jnp
. That's a good point, and it's better to do it before it becomes too complicated.
I assume you meant
jnp
. That's a good point, and it's better to do it before it becomes too complicated.
Yes! Just changed it
Waiting to see if the collaboration with PyMC3 and Numpyro happens before making the changes here.
This is to keep aligned with the Jax core and the documentation. From this issue: there has been a bit of a transition in the usage of JAX-flavored numpy from JAX is a drop-in replacement for numpy to JAX is a tool to use beside numpy.