Rebecca Chen
|
5ae2e79d43
|
Silence some pytype errors.
PiperOrigin-RevId: 520150523
|
2023-03-28 15:24:48 -07:00 |
|
Jake VanderPlas
|
760deb310e
|
Remove leading underscores in jax._src.numpy.util
|
2023-03-13 12:18:36 -07:00 |
|
Jake VanderPlas
|
c8c269f5f5
|
internal: avoid unused imports in lax_numpy
|
2023-03-08 10:29:04 -08:00 |
|
Peter Hawkins
|
8fb1fd318d
|
Replace jax._src.util.prod with math.prod.
math.prod() was added in Python 3.8, so we can assume it is always present.
PiperOrigin-RevId: 513011144
|
2023-02-28 12:41:00 -08:00 |
|
Roy Frostig
|
cb8dcce2fe
|
migrate more internal dependencies from jax.core to jax._src.core
PiperOrigin-RevId: 509736368
|
2023-02-14 23:01:11 -08:00 |
|
Jake VanderPlas
|
fd45035b21
|
[typing] add full annotations for lax_numpy setops
|
2022-10-03 12:52:28 -07:00 |
|
Peter Hawkins
|
ba557d5e1b
|
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
|
2022-09-22 12:27:19 -07:00 |
|
Jake VanderPlas
|
06c3857321
|
jnp.unique: improve error when run under JIT
|
2022-06-08 15:57:41 -07:00 |
|
Jake VanderPlas
|
466bea1662
|
lax_numpy: refactor set operations into separate private submodule
|
2022-03-21 09:38:11 -07:00 |
|