Skip to content
MPS gen models
Jupyter Notebook Python Shell
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.vscode
assets
experiments
mps
papers
test_datasets fixed bug in build_ghz May 1, 2019
.gitignore
README.md readme update May 2, 2019
bond-dimension-testing.ipynb some bond-dimension experiments May 6, 2019
cutoff-shaking-test-Copy1.ipynb
cutoff-shaking-test.ipynb
datagen.ipynb
do_hyperparam_selection.py updated training scripts May 9, 2019
do_training.py updated training scripts May 9, 2019
gen_mps_datasets.py same Jun 6, 2019
ghz-val-and-train.ipynb
ghz_test.ipynb added cutoff shaking; refactor train_from_filepath May 20, 2019
localupdate-multibasis-ghz-training.ipynb
localupdate-multibasis-training.ipynb add and rename sgd methods Apr 23, 2019
localupdate-randombasis-ghz.ipynb
localupdate-randombasis-prod.ipynb
localupdate-randombasis-training.ipynb
localupdate-singlebasis.ipynb
multibasis-nonuniform-test.ipynb minor Apr 18, 2019
multibasis-test.ipynb same Apr 17, 2019
notes.md
proto.ipynb
rand-U-fidelity-test.ipynb
sampling-demo.ipynb
singlebasis-test.ipynb
training-test.ipynb test-training Jun 26, 2019
validation-test.ipynb val+train script on GHZ May 8, 2019

README.md

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)
You can’t perform that action at this time.