jax

558ebb9f - Add Pallas Triton lowering for jax.lax.bitcast_convert_type.

Add Pallas Triton lowering for jax.lax.bitcast_convert_type. Only handles the case where operand type and target type have the same bitwidth. PiperOrigin-RevId: 696184251