Access the predicted values in the least squares loss. The normalized least squares loss uses an Adam-style rolling average normalization.
Access the predicted values in the least squares loss. The normalized
least squares loss uses an Adam-style rolling average normalization.

peterhj/neuralops,peterhj/neuralops