Roy Frostig f7731bf959 remove _const from public jax.lax module ...

Modify all internal call sites to use `jax._src.lax.lax._const`.

2022-03-07 12:26:25 -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

remove _const from public jax.lax module

2022-03-07 12:26:25 -08:00

linalg.py

remove _const from public jax.lax module

2022-03-07 12:26:25 -08:00

ndarray.py

lax_numpy.py: factor ndarray into its own module

2022-03-02 12:18:46 -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

Rollback of:

2022-03-01 19:39:31 -08:00

vectorize.py

djax: let make_jaxpr build dyn shape jaxprs

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