Peter Hawkins 78cfdd1b35 Add some more type annotations to lax_numpy.py. ...

These type annotations are of course mostly ignored because the pytype: skip-file comment, but they help readers if nothing else.

PiperOrigin-RevId: 555955257

2023-08-11 08:07:24 -07:00

..

control_flow

chore(docs): improve jax.lax.scan

2023-08-03 22:18:07 +02:00

__init__.py

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

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

ann.py

API: Remove NINF and PINF usages

2023-08-09 14:16:33 +02:00

convolution.py

Apply pyupgrade --py39-plus.

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

eigh.py

Improve the bisection strategy in eigh.

2023-05-04 18:01:23 -07:00

fft.py

Apply pyupgrade --py39-plus.

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

lax.py

Add some more type annotations to lax_numpy.py.

2023-08-11 08:07:24 -07:00

linalg.py

Add version guards after GPU tridiagonal solve change.

2023-08-11 06:41:05 -07:00

other.py

Apply pyupgrade --py39-plus.

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

parallel.py

Remove allowlist for multihost collectives.

2023-08-09 04:43:51 -07:00

qdwh.py

Use lower-case PEP 585 names for types.

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

slicing.py

lax.dynamic_slice: avoid negative index correction for unsigned indices

2023-07-25 13:09:09 -07:00

special.py

Split dtype argument from other arguments in special functions.

2023-03-25 11:41:14 -07:00

stack.py

Use lower-case PEP 585 names for types.

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

svd.py

Apply pyupgrade --py39-plus.

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

utils.py

Copybara import of the project:

2023-07-24 14:38:20 -07:00

windowed_reductions.py

Apply pyupgrade --py39-plus.

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