enerzyme.data.datatype# Functions is_int(k) is_rounded(k) is_atomic(k) requires_grad(k) is_idx(k) get_tensor_rank(k) is_target(k) is_target_uq(k) register_data_type(k, **type_info) is_grad(k)