Skip to content

Port Optimizers to Python

Vicentini Filippo requested to merge py_opt into v3.0

Created by: gcarleo

Rewrite of optimizers in Python.

In the process, I also a wrote a wrapper for PyTorch solvers:

import netket as nk
from torch.optim import SGD
....

# RBM Spin Machine
ma = nk.machine.RbmReal(alpha=1, hilbert=hi)

# Use a Torch optimizer 
op = nk.optimizer.Torch(ma, SGD, lr=0.01)

The same thing can be done for JAX solvers, thus removing the need for make_opt etc.

Closes #368 (closed)

Merge request reports