Got a chance to try out TorchStudio, a local IDE purpose built for PyTorch. While still in early development, it has an impressive feature list that includes analyzing datasets, building visual model graphs and experimenting and comparing local training jobs.