[PyTorch] Numerical Discrepancy Between torch.nn.attention.flex_attention and F.scaled_dot_product_attention — Expected or Bug?

Part of

Jetpack Compose Material 3 DatePicker animation glitch when