机器之心 (微信公众号)T2·85 pts
- Standard Softmax and Cross-Entropy implementations in PyTorch, JAX, and TensorFlow contain hidden numerical precision bugs
- These bugs can alter gradients during specific training phases, causing loss spikes or capability degradation
- The issue is fixable by adjusting computation methods without increasing floating-point precision or redesigning loss functions
- Gemini played a crucial role in discovering the bug and running extensive validation experiments