Jake VanderPlas e13c847e04 Index update operators: add scatter_apply()

2022-02-18 09:44:40 -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

Index update operators: add scatter_apply()

2022-02-18 09:44:40 -08:00

linalg.py

linalg_test: disable implicit rank promotion

2022-01-26 09:29:06 -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

DOC: add ability to document extra_params within _wraps

2022-02-10 16:54:57 -08:00

vectorize.py

djax: let make_jaxpr build dyn shape jaxprs

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