jax

e287f98c - Fix definition of qr primitive to return only the upper triangular part of r. (#2870)