atom_position_gradient kernel computes entropy-based gradient for
spacing_raw parameters. SGD decay toward uniform (lr=1e-3, every 50
steps) prevents atom positions from drifting. Concentrates atoms
where return distribution has mass.
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>