Next token after 'the': {'</s>': 0.1111111111111111, 'cat': 0.3333333333333333, 'dog': 0.2222222222222222, 'eats': 0.1111111111111111, 'sleeps': 0.1111111111111111, 'the': 0.1111111111111111}
Temperature 1.0: [0.665241, 0.244728, 0.090031]
Temperature 0.5: [0.866813, 0.11731, 0.015876]
Causal weights: [[1.0, 0.0], [0.3302384506733431, 0.6697615493266569]]
Attention output: [[10.0, 0.0], [3.302384506733431, 13.395230986533138]]
