jax

39d73a68 - Merge pull request #25276 from mattjj:remove-vestigial-reducing-transposes