Refactor array_api
namespace, relying more directly on jax.numpy
#246
Loading
array_api
namespace, relying more directly on jax.numpy
#246