tttc3 b1b56ea0b0 Enable pivoted QR on GPU via MAGMA. ...

Originally noted in #20282, this commit provides a GPU compatible
implementation of `geqp3` via MAGMA.

2025-02-12 16:12:42 +00: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

Enable pivoted QR on GPU via MAGMA.

2025-02-12 16:12:42 +00: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