New Paper Shows Gemini-Found Softmax Bug Fix Still Fails Property Tests

brandondamos · x · 2026-10-28

An arXiv paper (2610.36427) reproduces the recently discovered softmax numerical bug (found by Gemini) with property-based tests in Hypothesis, shows the proposed fix still fails an edge case, and presents a more robust softmax that passes the tests and supports causal attention masking — demonstrating property-based testing as a powerful tool for ML library numerics.

Original post →

More from Models

Models channel →