NA make kfac work in minsr

jeffminlin / vmcnet

Flexible, general-purpose VMC framework, built on JAX.

https://jeffminlin.github.io/vmcnet/

MIT License

24 stars 2 forks source link

NA make kfac work in minsr #129

Closed nilin closed 1 year ago

nilin commented 1 year ago

These changes KFAC work in the same branch as MinSR by placing the MinSR-related information in the "aux_data" (previously tuple) and "data" dictionaries without modifying the gradient information.