Visualizing pytorch's transformer feats

Hi there! Does anyone know a library to visualize pytorch’s tranformers?

1 Like

Here’s a couple packages I’ve used to visualize transformers. Both of them end up plotting details that I personally didn’t require for my use case, so you might have to tweak your plots as well.