Stable optimization of a tensor product variational state

We consider a variational problem for three-dimensional (3D) classical lattice models. We construct the trial state as a two-dimensional product of local variational weights that contain auxiliary variables. We propose a stable numerical algorithm for the maximization of the variational partition function per layer. The numerical stability and efficiency of the new method are examined through its application to the 3D using model.