jax
b13364fb - Minor improvements to doc for jax.nn.logsumexp.

Commit

326 days ago

Minor improvements to doc for jax.nn.logsumexp.

References

#392 - CI: 04/30/25 upstream sync

#518 - [0.6.0-UT] Adjust tolerance for reciprocal in unary primitives test

#519 - [0.6.0-UT] Fix Test Shard

#594 - Add CI workflow for Pytest ROCm

Author

carlosgmartin

carlosgmartin

Committer

charleshofer

charleshofer

Parents

Loading