Closed EiffL closed 4 years ago
jax.numpy was imported as jnp but then all the rest of the code was using np. so this PR fixes that
This solves issue #26
It's a tiny change and I've confirmed that it works on my machine, so I'm gonna go ahead and merge it
jax.numpy was imported as jnp but then all the rest of the code was using np. so this PR fixes that
This solves issue #26