jax

f9f06d42 - Merge pull request #37254 from samanklesaria:fix_triangular