jax.numpy

Imported 92 times

  1. 66 × import jax.numpy as jnp
  2. 14 × from jax import numpy as jnp
  3. 12 × import jax.numpy as np