MPS generative models.
models.py: defines the
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
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)