Skip to content

VMC driver and convenience functions in Python

Vicentini Filippo requested to merge github/fork/femtobit/py-driver into v2.1

Created by: femtobit

This is a work-in-progress demonstration of how the VMC driver can be implemented in Python now.

The example driver added largely has the same functionality as the current C++ one. In addition to the NetKet optimizers, it is possible to pass optimizers from jax.experimental.optimizers or any lambda function update computing p[i+1] = update(i, grad, p[i]).

Merge request reports