jax

491fcbb2 - floating point identity to inf