Peter Hawkins ac3cb6f954 Simplify mlir.dense_int_array. ...

The NumPy array conversion here is pointless and slightly slower than not doing it.

PiperOrigin-RevId: 647520922

2024-06-27 19:33:06 -07:00

..

control_flow

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04: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

Simplify mlir.dense_int_array.

2024-06-27 19:33:06 -07:00

linalg.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

other.py

Define lax.Precision directly in Python, rather than inheriting from a C++ type in jaxlib.

2024-03-04 14:01:31 -08:00

parallel.py

Fix Typos

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

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

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -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

Avoid a couple more uses of operator.attrgetter(), since it's unnecessarily slow compared to writing a function directly.

2024-02-23 09:20:56 -08:00

windowed_reductions.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00