Matthew Johnson f0392a1535 fix grad(logsumexp) to produce 0s where where is False

2025-01-08 23:38:06 +00: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

Reverts 6cf09f8c24c67ff650b95d174501fff3cb59db0d

2024-10-04 13:56:27 -07:00

special.py

fix grad(logsumexp) to produce 0s where where is False

2025-01-08 23:38:06 +00:00