[skumar17]

Unfortunately it might be a bit challenging as there’s a nontrivial amount of extra computation we do for the viz, but it’s probably possible?

[skumar17]

The attention demo code is in the /attention_demo directory if you want to try it on your own messages too :)