func(Callable): Torch function for which we want to provide a sharded
            implementation (ex: torch.nn.functional.linear)
    ©