mirror of
https://github.com/google-deepmind/deepmind-research.git
synced 2025-12-10 06:51:04 +08:00
f905943c13
This is deprecated as of https://github.com/google/jax/pull/15263: most users will never need to use ShapedArray directly, and so having it exposed in the top-level public namespace causes undue confusion. PiperOrigin-RevId: 520846838
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