Jake VanderPlas 8d17cce80e Add JIT-compatible version of jnp.nonzero

2021-04-20 09:18:49 -07:00

..

__init__.py

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

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

fft.py

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

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

lax_numpy.py

Add JIT-compatible version of jnp.nonzero

2021-04-20 09:18:49 -07:00

linalg.py

[JAX] Move contents of jax/dtypes.py to jax/_src/dtypes.py.

2021-04-07 19:35:51 -07:00

polynomial.py

Run doctest on all source files except jax2tf

2021-04-05 10:39:59 -07:00

util.py

docs: tweaks to make docs build on Python 3.8

2021-04-01 15:50:15 -07:00

vectorize.py

Move jax.api to jax._src.api.

2021-04-13 09:43:24 -07:00