jax authors 39786c6410 Merge pull request #9394 from jakevdp:pre-commit-versions ...

PiperOrigin-RevId: 425681158

2022-02-01 11:56:48 -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

Merge pull request #9394 from jakevdp:pre-commit-versions

2022-02-01 11:56:48 -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

maint: update pre-commit package versions & fix new mypy errors

2022-01-31 13:39:11 -08:00

vectorize.py

djax: let make_jaxpr build dyn shape jaxprs

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