MPS generative models.
models.py
: defines theMPS
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
- halt the proliferation of
_utils
modules MPS sampling methodsBuild some large-N datasetspackage training code from notebooksAdd 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)