Named tensors with first-class dimensions for PyTorch - View it on GitHub