Peter Hawkins 3d5784a343 Don't wrap singleton ir.Types during HLO lowering. ...

This is similar to https://github.com/google/jax/pull/22211, but for MLIR types instead of MLIR values.

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

..

control_flow

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

2024-07-08 12:24:45 -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

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

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

linalg.py

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

2024-07-01 16:11:00 -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

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

2024-07-01 16:11:00 -04: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

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

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

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