jax.dtypes.result_type# jax.dtypes.result_type(*args, return_weak_type_flag=False)# Convenience function to apply JAX argument dtype promotion. Parameters: return_weak_type_flag (bool) – if True, then return a (dtype, weak_type) tuple. If False, just return dtype args (Any) – Return type: Union[dtype, tuple[dtype, bool]] Returns: dtype or (dtype, weak_type) depending on the value of the return_weak_type argument.