For some reason PyCharm gets torch.einsum signature wrong, and every call to torch.einsum triggers Type Checker failure
def einsum(equation, *operands):
r"""einsum(equation, *operands) -> Tensor
More generally, "torch" namespace is hard because a lot of functions support automatic conversion from numpy arrays, which is not properly documented in their function signature. Is there a way to disable type checking for an entire namespace?