Roy Frostig a60ead6fd1 enable partitionable threefry by default ...

PiperOrigin-RevId: 715242560

2025-01-13 22:46:24 -08: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

Fix false positive debug_nans error caused by NaNs that are properly handled in jax.scipy.stats.gamma

2024-11-21 10:33:29 -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 CPU devices.

2025-01-09 20:44:45 +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