# Cool down the chaos to find the truth

# Or heat it up to dream

# Scale before you normalize

probs = jax.nn.softmax(logits / )

# Incorrect answer

# Try again