Closed sidravi1 closed 3 years ago
jax and numpyro both use jnp for jax.numpy and np for numpy. To avoid confusion, we should switch to their convention:
jax
numpyro
jnp
jax.numpy
np
numpy
Need to go through the entire codebase, change these lines:
import jax.numpy as np -> import jax.numpy as jnp import numpy as onp -> import numpy as np
import jax.numpy as np
import jax.numpy as jnp
import numpy as onp
import numpy as np
and replace all references of np with jnp and onp with np.
onp
Hi! I'm going to work on this. Could you please assign me this issue? Thanks beforehand :)
Fixed in #73, that was fast!
jax
andnumpyro
both usejnp
forjax.numpy
andnp
fornumpy
. To avoid confusion, we should switch to their convention:Need to go through the entire codebase, change these lines:
import jax.numpy as np
->import jax.numpy as jnp
import numpy as onp
->import numpy as np
and replace all references of
np
withjnp
andonp
withnp
.