jax authors be543f020d Merge pull request #17041 from mtsokol:update-ninf-usage ...

PiperOrigin-RevId: 555185385

2023-08-09 09:26:12 -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

Move functions out of xla.py closer to their users.

2023-08-08 14:40:42 -07:00

linalg.py

schur: fix broken jvp rule

2023-06-30 02:30:25 -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