Pytorch pypi. Visualization includes tensors, modules, torch.