Peter Hawkins 2c2f4033cc Move contents of jax.lib to jax._src.lib. ...

Add shim libraries for functions exported from jax.lib that other code seems to use in practice.

PiperOrigin-RevId: 398471863

2021-09-23 06:33:55 -07:00

..

__init__.py

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

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

fft.py

Move contents of jax.lib to jax._src.lib.

2021-09-23 06:33:55 -07:00

lax_numpy.py

Merge pull request #7966 from jakevdp:faster-conv

2021-09-21 13:33:33 -07:00

linalg.py

Switch internal users of jax.ops.index_... to use x.at[x].set() APIs.

2021-09-13 19:48:29 -04:00

polynomial.py

Switch internal users of jax.ops.index_... to use x.at[x].set() APIs.

2021-09-13 19:48:29 -04:00

util.py

docs: tweaks to make docs build on Python 3.8

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

vectorize.py

DOC: many small fixes

2021-08-04 16:55:13 -07:00