Skip to content

Testing using AI and ML techniques to solve quantum systems

License

Notifications You must be signed in to change notification settings

tilakrayal/MLQM

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DOI

MLQM

MLQM stands for "Machine Learning Quantum Montecarlo". This repository contains tools to perform variational monte carlo for nuclear physics, though there are some additional Hamiltonians implemented as development tools and cross checks: the harmonic oscillator and the Hydrogen atom.

Requirements

The requirements to run this code are:

  • python > 3.6
  • tensorflow > 2.X (not TF1 compatible)
  • hydra-core > 1.0 (for configuration)
  • horovod (for scaling and multi-node running)

There is no installation step, it's expect that once you have the requirements install you can begin running immediately.

Configuration and Running

The main executable is bin/stochastic_reconfiguration.py. You can execute it with:

python bin/stochastic_reconfiguration.py run_id=MyTestRun 

Most parameters have reasonable defaults, which you can change in configuration files (in config/) or override on the command line:

python bin/stochastic_reconfiguration.py run_id=deuteron nparticles=2 iterations=500 optimizer=AdaptiveDelta [... other argument overrides] 

Computational Performance

This software is compatible with both CPUs and GPUs through Tensorflow. It has good weak and strong scaling performance:

Scaling performance for 4He on A100 GPUs (ThetaGPU@ALCF)

The software also has good scaling performance with increasing number of nucleons:

Nucleon scaling performance on A100 GPUs (ThetaGPU@ALCF)

Reference

If you use this software, please reference our publication on arxiv

About

Testing using AI and ML techniques to solve quantum systems

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%