Jake VanderPlas eced12d89b Finalize deprecation of lax.linalg positional args ...

PiperOrigin-RevId: 629581163

2024-04-30 17:56:18 -07:00

..

control_flow

Let xs=None by default in lax.scan.

2024-04-23 17:26:23 -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

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

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

convolution.py

Emit dense arrays for StableHLO ops migrating to dense arrays

2024-01-24 16:41:37 -08: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

Merge branch 'main' into refs-in-vjps

2024-04-12 15:25:37 -04:00

linalg.py

Finalize deprecation of lax.linalg positional args

2024-04-30 17:56:18 -07: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

Minor typo fix in docstring jax.lax.psum

2024-04-12 13:12:00 +01:00

qdwh.py

Improve performance of SVD when batched by avoiding several cond() constructs.

2024-04-26 15:00:50 -07:00

slicing.py

Correct a name in ScatterDimensionNumbers docstring

2024-04-15 10:36:24 +00: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

Improve performance of SVD when batched by avoiding several cond() constructs.

2024-04-26 15:00:50 -07:00

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

Split name_stack out of mlir.ModuleContext.

2024-02-20 07:17:23 -08:00