George Necula ffd2b00516 Add concretization error check in core.min_dim and core.max_dim ...

Fixes: #22751

2024-08-01 07:27:35 +02:00

..

control_flow

remove named_shapes (since xmap is now gone)

2024-07-25 00:54:50 +00: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

Bump minimum jaxlib version to v0.4.30.

2024-06-18 12:35:08 -04:00

convolution.py

Clean up version switches from dense array migration

2024-05-28 10:58:51 -07:00

eigh.py

Return early from eigh for small matrices. This was accidentally removed cl/577222219.

2024-04-22 15:10:38 -07:00

fft.py

Bumped the minimum jaxlib to 0.4.23

2024-04-13 08:18:33 +01:00

lax.py

Add concretization error check in core.min_dim and core.max_dim

2024-08-01 07:27:35 +02:00

linalg.py

Activate Cholesky Factorization Kernel to XLA's FFI

2024-07-25 09:59:28 -07:00

other.py

Bumped jaxlib version mypy uses on the CI

2024-07-26 11:22:39 +01:00

parallel.py

remove pdot and xeinsum (since xmap is gone)

2024-07-25 21:19:17 +00:00

qdwh.py

Get rid of the is_hermitian argument for lax.qdwh. If it was known that H was also positive semi-definite, the polar decomposition would be I*H. But for indefinite H, the QDWH algorithm does not differ from the general case for Hermitian inputs.

2024-06-13 15:33:49 -07:00

slicing.py

Don't wrap singleton ir.Types during HLO lowering.

2024-07-08 12:24:45 -04:00

special.py

Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result.

2023-11-17 11:47:14 -08:00

stack.py

Use lower-case PEP 585 names for types.

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

svd.py

Fix Typos

2024-06-17 13:55:46 +05:30

utils.py

remove named_shapes (since xmap is now gone)

2024-07-25 00:54:50 +00:00

windowed_reductions.py

Don't wrap singleton ir.Values with tuples during HLO lowering.

2024-07-01 16:11:00 -04:00