Support for Transformer based NMT Attention Mechenisms

Hi,

This viz tool looks good. However I have attentions files from Transformer based models which I'm using for Machine Translation. Is there support for this ? Any other links or pointers also helpful 

Thanks !

Mohammed Ayub