Software Engineer working on PyTorch at NVIDIA