Skip to content

briantimar/mps-gm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mps-gm

MPS generative models.

  • models.py: defines the MPS model.
  • old_models.py: a few other MPS-type models, deprecated.
  • qtools.py: utilities for dealing with measurements in multiple bases.
  • qutip_utils.py: utilities for dealing with qutip states.
  • tools.py: miscellaneous utils related to discrete probability distributions.
  • utils.py: code for SVD updates and other PyTorch related.
  • test.py: an (incomplete) collection of unit tests

TODO

  • halt the proliferation of _utils modules
  • MPS sampling methods
  • Build some large-N datasets
  • package training code from notebooks
  • Add GPU support (this is significantly slower than CPU on datasets I've tried so far -- tensors have to be moved to cpu for every SVD)