Hmm there are couple of them.  You can try tensorboard X which is a pytorch version of tensorboard.  There is another one which is nice and group layers not to have a huge graph. The rest of them are very simple and graphs are usually illegible.