Currently a PhD student in computer science. New to PyTorch.