Why not use torch.compile()?