Jake VanderPlas 8f72454bdf Add internal jax.lax.asarray utility

2023-03-30 10:21:55 -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

[JAX] Delete _DeviceArray and DeviceArray.

2023-03-29 15:07:14 -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

[JAX] Delete _DeviceArray and DeviceArray.

2023-03-29 15:07:14 -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

Add internal jax.lax.asarray utility

2023-03-30 10:21:55 -07:00

setops.py

Use cast instead of pytype suppression in setops.py.

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

ufuncs.py

Add internal jax.lax.asarray utility

2023-03-30 10:21:55 -07:00

util.py

Add internal jax.lax.asarray utility

2023-03-30 10:21:55 -07:00

vectorize.py

avoid unnecessary broadcasting in jax.numpy.vectorize

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