Tensor Comprehensions in PyTorch
pytorch.org