Jake VanderPlas ed2550999f implement jnp.copy

2022-03-01 11:56:36 -08:00

..

__init__.py

Move jax.numpy internals into jax._src.numpy.

2020-10-16 20:35:19 -04:00

fft.py

Use imports relative to the jax package consistently, rather than .-relative imports.

2021-11-24 07:48:29 -08:00

lax_numpy.py

implement jnp.copy

2022-03-01 11:56:36 -08:00

linalg.py

linalg_test: disable implicit rank promotion

2022-01-26 09:29:06 -08:00

ndarray.py

Factor-out pieces of lax_numpy.py

2022-03-01 09:40:37 -08:00

polynomial.py

Use imports relative to the jax package consistently, rather than .-relative imports.

2021-11-24 07:48:29 -08:00

ufuncs.py

Merge pull request #9724 from jakevdp:refactor-lax-numpy

2022-03-01 11:33:39 -08:00

util.py

Factor-out pieces of lax_numpy.py

2022-03-01 09:40:37 -08:00

vectorize.py

djax: let make_jaxpr build dyn shape jaxprs

2022-02-01 00:10:21 -08:00