Skip to content

heikoburau/ANNonGPU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

176 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ANNonGPU

This library implements the algorithm of Carleo and Troyer for representing quantum many-body wavefunction using various artificial neural network (ANN) architectures on the GPU.

Installation

cd ANNonGPU
cmake . -DCMAKE_INSTALL_PREFIX=<e.g. $HOME/.local>
make install

(for python interface only)
cd pyANNonGPU
python setup.py install

Features

  • Supported network architectures: RBM, deep CNN, vCN
  • Markov-Chain Monte-Carlo sampling and exact summation
  • Arbitrary quantum operators
  • Generic interface for gradient descent optimization
  • Stochastic reconfiguration using TDVP

About

Implementation of the Stochastic Reconfiguration algorithm on the GPU using C++/CUDA.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors