jax

6481bb28 - Remove primitive_transpose registrations for primitives that have a transpose_fancy rule defined