jax

05b4ac80 - Merge pull request #32493 from mattjj:newstyle-hiprimitive2