Jake VanderPlas f8e18e9a00 [x64] minor weak_type changes to linalg.py

2021-12-07 16:27:29 -08:00

..

optimize

Use imports relative to the jax package consistently, rather than .-relative imports.

2021-11-24 07:48:29 -08:00

sparse

[x64] minor weak_type changes to linalg.py

2021-12-07 16:27:29 -08:00

stats

stats.multivariate_normal: support broadcasted inputs

2021-10-19 16:58:36 -07:00

__init__.py

Move most of the implementation of jax.scipy into jax._src.scipy.

2020-10-16 17:04:25 -04:00

eigh.py

Document jax.lax.Precision

2021-11-08 14:15:31 -08:00

fft.py

Cleanup: remove duplicate canonicalize_axis utility

2021-11-23 16:54:02 -08:00

linalg.py

Add @jit decorators to jax.numpy.linalg and jax.scipy.linalg.

2021-09-24 15:52:11 -04:00

ndimage.py

Implement 'reflect' and 'mirror' padding modes for scipy.ndimage.map_coordinates

2021-09-07 16:05:35 +02:00

signal.py

convolutions: use flip() to clean up reverse-indexing

2021-09-21 08:49:32 -07:00

special.py

Tighten up dtypes across the package

2021-10-29 13:50:30 -07:00