Teaching the Rayleigh–Ritz variational principle with neural-network trial wave functions
We present a computational module for upper-level undergraduate quantum mechanics and computational physics courses in which neural networks serve as variational trial wave functions for the Rayleigh–Ritz method. Spatial derivatives are evaluated using automatic differentiation. The infinite square well and harmonic os...