torch.compile() experiments