Jan Naumann e03fe3a06d Implement SVD algorithm based on QR for CPU targets ...

In a recent jax release the SvdAlgorithm parameter has been added
to the jax.lax.linalg.svd function. Currently, for CPU targets
still only the divide and conquer algorithm from LAPACK is
supported (gesdd).

This commits adds the functionality to select the QR based
algorithm on CPU as well. Mainly it addes the wrapper code
to call the gesvd function of LAPACK using the FFI interface.

Signed-off-by: Jan Naumann <j.naumann@fu-berlin.de>

2025-02-22 15:24:57 +01:00

..

cluster

DOC: Improve remaining jax.scipy docstrings

2024-05-02 10:29:00 -07:00

interpolate

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

optimize

Run pyupgrade --py310-plus.

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

sparse

Removed various ununsed functions

2024-08-01 11:18:19 +01:00

spatial

Add Rotation return type hint to Rotation.__mul__()

2024-12-11 12:48:02 +01:00

stats

refactor: move scalar type defs out of lax_numpy.py

2025-02-06 14:48:10 -08:00

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

fft.py

enable partitionable threefry by default

2025-01-13 22:46:24 -08:00

integrate.py

DOC: Improve remaining jax.scipy docstrings

2024-05-02 10:29:00 -07:00

linalg.py

Implement SVD algorithm based on QR for CPU targets

2025-02-22 15:24:57 +01:00

ndimage.py

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

signal.py

Fixed some type errors under pyright

2024-09-05 09:56:38 +01:00

special.py

Add sph_harm_y to jax.scipy.special and deprecate sph_harm.

2025-01-09 12:53:00 -05:00