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

Sign in to join the discussion — your thebay.events account works here.