PyTorch jobs in Stockholm on