Files
ml_lib/metrics/attention_rollout.py