Hi,
As one of the person that originally created crayon, I do not actively use it anymore (because I don’t need this kind of visualization). And so there are no new features added to it by the original authors.
I don’t know what is the status of tensorboard-pytorch though.