Yash Katariya 4fc3518e5f Make checkify tests pass with Array and add methods on Array that are present on DA. ...

PiperOrigin-RevId: 468058909

2022-08-16 16:52:06 -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

Make checkify tests pass with Array and add methods on Array that are present on DA.

2022-08-16 16:52:06 -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

Remove _ prefix from functions in jax._src.dtypes.

2022-08-12 12:51:09 +00:00

reductions.py

Remove _ prefix from functions in jax._src.dtypes.

2022-08-12 12:51:09 +00:00

setops.py

jnp.unique: improve error when run under JIT

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

ufuncs.py

Merge pull request #11857 from jakevdp:fix-bool-args

2022-08-12 11:45:35 -07:00

util.py

Merge pull request #11857 from jakevdp:fix-bool-args

2022-08-12 11:45:35 -07:00

vectorize.py

feat: refactor code using pyupgrade

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