This is very cool. It's simpler than Bertviz for understanding inference and surface level and a good starting people for new learners as well.