Parker Schuh 10289390f3 Decouple ShardedDeviceArray from _DeviceArray

2021-02-25 13:11:32 -08: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

Decouple ShardedDeviceArray from _DeviceArray

2021-02-25 13:11:32 -08:00

linalg.py

Support empty inputs in jnp.linalg.svd() and jnp.linalg.pinv()

2021-01-13 10:39:00 -08:00

polynomial.py

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

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

util.py

DOC: fix update_numpydoc to prevent deleting docstring content

2021-02-22 16:45:40 -08:00

vectorize.py

[JAX] Move jax.util to jax._src_util.

2021-01-11 14:21:07 -08:00