Is there any good package to visualize attention weights on words in text?

Hi, I am not looking for something like this

but something that colors the text it self.

You can save attention weights, then use matplotlib visualize.