George Necula 5983d385da [dynamic-shapes] Expand the handling of dynamic shapes for reshape and iota. ...

Also add more tests.

2022-07-05 12:14:15 +03:00

..

__init__.py

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

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

fft.py

[x64] make fft functionality compatible with strict dtype promotion

2022-06-15 10:10:44 -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

[dynamic-shapes] Expand the handling of dynamic shapes for reshape and iota.

2022-07-05 12:14:15 +03:00

linalg.py

[x64] make linalg functions & tests compatible with strict dtype promotion

2022-06-16 10:32:20 -07:00

ndarray.py

Make jnp.array return jax.Array. Add input and result handlers for jax.Array. Also added tests for add under jit.

2022-06-24 10:05:06 -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

[x64] make jnp.modf() compatible with strict dtype promotion

2022-06-03 15:23:06 -07:00

util.py

Use equality rather than identity when checking for float0

2022-06-23 11:46:20 -07:00

vectorize.py

feat: refactor code using pyupgrade

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