We present a neural network wave function framework for solving non-Abelian lattice gauge theories in a continuous group representation. Using a combination of SU(2) equivariant neural networks alongside an SU(2) invariant, physics-inspired ansatz, we learn a parametrization of the ground state wave function of SU(2) lattice gauge theory in 2+1 and 3+1 dimensions. Our method, performed in the Hamiltonian formulation, has a straightforward generalization to SU(N). We benchmark our approach against a solely invariant ansatz by computing the ground state energy, demonstrating the need for bespoke gauge equivariant transformations. We evaluate the Creutz ratio and average Wilson loop, and obtain results in strong agreement with perturbative expansions. Our method opens up an avenue for studying lattice gauge theories beyond one dimension, with efficient scaling to larger systems, and in a way that avoids both the sign problem and any discretization of the gauge group.

Variational techniques have long been at the heart of atomic, solid-state, and many-body physics. They have recently extended to quantum and classical machine learning, providing a basis for representing quantum states via neural networks. These methods generally aim to minimize the energy of a given ansatz, though open questions remain about the expressivity of quantum and classical variational ansätze. The connection between variational techniques and quantum computing, through variational quantum algorithms, offers opportunities to explore the quantum complexity of classical methods. We demonstrate how the concept of non-stabilizerness, or magic, can create a bridge between quantum information and variational techniques and we show that energy accuracy is a necessary but not always sufficient condition for accuracy in non-stabilizerness. Through systematic benchmarking of neural network quantum states, matrix product states, and variational quantum methods, we show that while classical techniques are more accurate in non-stabilizerness, not accounting for the symmetries of the system can have a severe impact on this accuracy. Our findings form a basis for a universal expressivity characterization of both quantum and classical variational methods.