Jake VanderPlas 7d6a134f4e logsumexp: use NumPy 2.0 convention for complex sign

2024-01-16 16:15:06 -08:00

..

__init__.py

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

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

scatter.py

[shape_poly] Update the jax.ops.segment{max|...} to with with shape polymorphism

2023-12-19 12:02:39 +02:00

special.py

logsumexp: use NumPy 2.0 convention for complex sign

2024-01-16 16:15:06 -08:00