Jake VanderPlas a5737f82af custom prng: remove stackable override for jnp.concatenate

2023-04-24 12:26:58 -07:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

array_methods.py

[djax] handle simple reshapes and size-0 checks

2023-04-21 19:20:48 -07:00

fft.py

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -07:00

index_tricks.py

Improve type of jnp.mgrid[...].

2023-03-13 17:46:03 -07:00

lax_numpy.py

custom prng: remove stackable override for jnp.concatenate

2023-04-24 12:26:58 -07:00

linalg.py

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -07:00

polynomial.py

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -07:00

reductions.py

Change np.prod->math.prod

2023-04-13 11:48:11 -07:00

setops.py

Use cast instead of pytype suppression in setops.py.

2023-03-29 11:17:00 -07:00

ufuncs.py

Avoid imports from the public jax.* namespace in more places internally.

2023-04-04 11:41:40 -07:00

util.py

custom prng: remove stackable override for jnp.concatenate

2023-04-24 12:26:58 -07:00

vectorize.py

avoid unnecessary broadcasting in jax.numpy.vectorize

2023-03-07 11:56:34 +01:00