jax

ee3a071b - Fix: always canonicalize result_type