Jake VanderPlas 1a327f1ddd jnp.linalg.matrix_rank: support stacks of matrices

2022-08-08 11:16:34 -07:00

..

__init__.py

Move jax.numpy internals into jax._src.numpy.

2020-10-16 20:35:19 -04:00

fft.py

Do not normalize FFT by a constant "1" if no normalization is provided (i.e., norm is None).

2022-07-07 11:54:39 -07:00

index_tricks.py

[x64] make jnp.mgrid compatible with strict dtype promotion

2022-06-01 10:21:49 -07:00

lax_numpy.py

Add dtype arg to jnp.concatenate and update tests

2022-08-01 15:48:40 -07:00

linalg.py

jnp.linalg.matrix_rank: support stacks of matrices

2022-08-08 11:16:34 -07:00

ndarray.py

Add dlpack support to device_array and jax.numpy

2022-07-15 17:31:11 -07:00

polynomial.py

jnp.roots: better support for computation under JIT

2022-06-23 14:48:53 -07:00

reductions.py

jnp.average: support keepdims argument

2022-06-28 10:55:55 -07:00

setops.py

jnp.unique: improve error when run under JIT

2022-06-08 15:57:41 -07:00

ufuncs.py

DOC: ensure that _wraps() generates correct links to wrapped functions

2022-07-21 11:12:35 -07:00

util.py

DOC: ensure that _wraps() generates correct links to wrapped functions

2022-07-21 11:12:35 -07:00

vectorize.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00