Jake VanderPlas 080d70e58a jax.numpy: add where and initial arguments to nan reductions

2022-01-25 09:17:07 -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

jax.numpy: add where and initial arguments to nan reductions

2022-01-25 09:17:07 -08:00

linalg.py

[x64] minor weak_type changes to linalg.py

2021-12-07 16:27:29 -08:00

polynomial.py

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

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

util.py

Fix auto-generated docstrings for JIT-compiled functions

2022-01-24 09:19:51 -08:00

vectorize.py

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

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