jax

09325d92 - Update internal unop primitive helper to pass kwargs to dtype rule.

Update internal unop primitive helper to pass kwargs to dtype rule. To be consistent with other rule registration helpers, `unop_dtype_rule` should pass through its kwargs to the `result_dtype` callable. PiperOrigin-RevId: 729483613