kfac_ferminet_alpha

Accompanying code for Better, Faster Fermionic Neural Networks

All package requirements are listed in requirements.txt.

Contributing

This is purely research code, provided with no further intentions of support or any guarantees of backward compatibility.

Installation

git clone git@github.com:deepmind/deepmind-research.git
pip install deepmind_research/kfac_ferminet_alpha/

Usage

You can find examples of how to use the codebase through the FermiNet project.

We also provide an example training script.

Reference

Better, Faster Fermionic Neural Networks

James S. Spencer, David Pfau, Aleksandar Botev, and W. M. C. Foulkes.

URL: https://arxiv.org/abs/2011.07125.

Optimizing Neural Networks with Kronecker-factored Approximate Curvature

James Martens, Roger Grosse

URL: https://arxiv.org/abs/1503.05671