Sergei Lebedev 2f70ae700a Migrate another subset of internal modules to use state objects ...

The motivation here is to gradually replace all dynamic lookups on `jax.config`
with statically-typed state objects, which are more type checker/IDE friendly.

This is a follow up to #18008.

PiperOrigin-RevId: 572587137

2023-10-11 08:46:06 -07:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

array_methods.py

Make jax._src.typing.DTypeLike more strictly defined

2023-10-10 09:01:19 -07:00

fft.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

index_tricks.py

Apply pyupgrade --py39-plus.

2023-07-21 14:49:44 -04:00

lax_numpy.py

Migrate another subset of internal modules to use state objects

2023-10-11 08:46:06 -07:00

linalg.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

polynomial.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

reductions.py

Make jax._src.typing.DTypeLike more strictly defined

2023-10-10 09:01:19 -07:00

setops.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00

ufunc_api.py

jnp.frompyfunc: fix .at() edge case

2023-10-09 11:24:25 -07:00

ufuncs.py

jnp.bitwise_count: fix issue with unsigned ints

2023-10-05 14:00:31 -07:00

util.py

Migrate a subset of internal modules to use state objects

2023-10-09 07:29:53 -07:00

vectorize.py

Use lower-case PEP 585 names for types.

2023-06-23 15:12:14 -07:00