Implementation in PyTorch of the self-attention neural ansatz for electronic structure calculations introduced in the paper “A Self-attention Ansatz for ab initio Quantum Chemistry” from Google Deep Mind.
training quantum-mechanics torch transformer computational-physics mlp attention-mechanism wandb google-deepmind natural-gradient wave-equation-solver kfca computational-quantum-mechanics
-
Updated
Jan 14, 2026 - TeX