├── .flake8 ├── .github └── workflows │ └── test.yml ├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── VERSION.txt ├── dev_requirements.txt ├── docs ├── .buildinfo ├── .nojekyll ├── Makefile ├── _config.yml ├── _modules │ ├── index.html │ └── torchmps.html ├── _sources │ ├── index.rst.txt │ └── source │ │ ├── README.rst.txt │ │ ├── adaptive.rst.txt │ │ ├── basic_usage.rst.txt │ │ ├── citing.rst.txt │ │ ├── functional.rst.txt │ │ ├── intro.rst.txt │ │ └── software.rst.txt ├── _static │ ├── alabaster.css │ ├── basic.css │ ├── classic.css │ ├── custom.css │ ├── doctools.js │ ├── documentation_options.js │ ├── file.png │ ├── jquery-3.5.1.js │ ├── jquery.js │ ├── language_data.js │ ├── minus.png │ ├── plus.png │ ├── pygments.css │ ├── searchtools.js │ ├── sidebar.js │ ├── underscore-1.12.0.js │ └── underscore.js ├── conf.py ├── genindex.html ├── index.html ├── index.rst ├── make.bat ├── objects.inv ├── py-modindex.html ├── search.html ├── searchindex.js └── source │ ├── adaptive.html │ ├── adaptive.rst │ ├── basic_usage.html │ ├── basic_usage.rst │ ├── citing.html │ ├── citing.rst │ ├── functional.html │ ├── functional.rst │ ├── intro.html │ ├── intro.rst │ ├── software.html │ └── software.rst ├── requirements.txt ├── setup.py ├── torchmps ├── __init__.py ├── contractables.py ├── embeddings.py ├── mps_base.py ├── prob_mps.py ├── tests │ ├── __init__.py │ ├── benchmarks │ │ ├── __init__.py │ │ ├── test_benchmark_mps_base.py │ │ └── test_benchmark_prob_mps.py │ ├── old_tests │ │ ├── custom_feature_map.py │ │ ├── dynamic_mps_basic.py │ │ ├── mat_region_open_periodic_bcs.py │ │ ├── static_mps_basic.py │ │ ├── svd_flex.py │ │ └── ti_mps_basic.py │ ├── test_embeddings.py │ ├── test_mps_base.py │ ├── test_prob_mps.py │ ├── test_utils2.py │ └── utils_for_tests.py ├── torchmps.py ├── utils.py └── utils2.py └── train_script.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/README.md -------------------------------------------------------------------------------- /VERSION.txt: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /dev_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/dev_requirements.txt -------------------------------------------------------------------------------- /docs/.buildinfo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/.buildinfo -------------------------------------------------------------------------------- /docs/.nojekyll: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_config.yml -------------------------------------------------------------------------------- /docs/_modules/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_modules/index.html -------------------------------------------------------------------------------- /docs/_modules/torchmps.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_modules/torchmps.html -------------------------------------------------------------------------------- /docs/_sources/index.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/index.rst.txt -------------------------------------------------------------------------------- /docs/_sources/source/README.rst.txt: -------------------------------------------------------------------------------- 1 | .. mdinclude:: ../../README.md 2 | -------------------------------------------------------------------------------- /docs/_sources/source/adaptive.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/source/adaptive.rst.txt -------------------------------------------------------------------------------- /docs/_sources/source/basic_usage.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/source/basic_usage.rst.txt -------------------------------------------------------------------------------- /docs/_sources/source/citing.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/source/citing.rst.txt -------------------------------------------------------------------------------- /docs/_sources/source/functional.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/source/functional.rst.txt -------------------------------------------------------------------------------- /docs/_sources/source/intro.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/source/intro.rst.txt -------------------------------------------------------------------------------- /docs/_sources/source/software.rst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_sources/source/software.rst.txt -------------------------------------------------------------------------------- /docs/_static/alabaster.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/alabaster.css -------------------------------------------------------------------------------- /docs/_static/basic.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/basic.css -------------------------------------------------------------------------------- /docs/_static/classic.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/classic.css -------------------------------------------------------------------------------- /docs/_static/custom.css: -------------------------------------------------------------------------------- 1 | /* This file intentionally left blank. */ 2 | -------------------------------------------------------------------------------- /docs/_static/doctools.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/doctools.js -------------------------------------------------------------------------------- /docs/_static/documentation_options.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/documentation_options.js -------------------------------------------------------------------------------- /docs/_static/file.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/file.png -------------------------------------------------------------------------------- /docs/_static/jquery-3.5.1.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/jquery-3.5.1.js -------------------------------------------------------------------------------- /docs/_static/jquery.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/jquery.js -------------------------------------------------------------------------------- /docs/_static/language_data.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/language_data.js -------------------------------------------------------------------------------- /docs/_static/minus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/minus.png -------------------------------------------------------------------------------- /docs/_static/plus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/plus.png -------------------------------------------------------------------------------- /docs/_static/pygments.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/pygments.css -------------------------------------------------------------------------------- /docs/_static/searchtools.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/searchtools.js -------------------------------------------------------------------------------- /docs/_static/sidebar.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/sidebar.js -------------------------------------------------------------------------------- /docs/_static/underscore-1.12.0.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/underscore-1.12.0.js -------------------------------------------------------------------------------- /docs/_static/underscore.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/_static/underscore.js -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/genindex.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/genindex.html -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/objects.inv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/objects.inv -------------------------------------------------------------------------------- /docs/py-modindex.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/py-modindex.html -------------------------------------------------------------------------------- /docs/search.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/search.html -------------------------------------------------------------------------------- /docs/searchindex.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/searchindex.js -------------------------------------------------------------------------------- /docs/source/adaptive.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/adaptive.html -------------------------------------------------------------------------------- /docs/source/adaptive.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/adaptive.rst -------------------------------------------------------------------------------- /docs/source/basic_usage.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/basic_usage.html -------------------------------------------------------------------------------- /docs/source/basic_usage.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/basic_usage.rst -------------------------------------------------------------------------------- /docs/source/citing.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/citing.html -------------------------------------------------------------------------------- /docs/source/citing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/citing.rst -------------------------------------------------------------------------------- /docs/source/functional.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/functional.html -------------------------------------------------------------------------------- /docs/source/functional.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/functional.rst -------------------------------------------------------------------------------- /docs/source/intro.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/intro.html -------------------------------------------------------------------------------- /docs/source/intro.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/intro.rst -------------------------------------------------------------------------------- /docs/source/software.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/software.html -------------------------------------------------------------------------------- /docs/source/software.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/docs/source/software.rst -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch >= 1.8.1 2 | opt_einsum >= 3.3.0 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/setup.py -------------------------------------------------------------------------------- /torchmps/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/__init__.py -------------------------------------------------------------------------------- /torchmps/contractables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/contractables.py -------------------------------------------------------------------------------- /torchmps/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/embeddings.py -------------------------------------------------------------------------------- /torchmps/mps_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/mps_base.py -------------------------------------------------------------------------------- /torchmps/prob_mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/prob_mps.py -------------------------------------------------------------------------------- /torchmps/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchmps/tests/benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchmps/tests/benchmarks/test_benchmark_mps_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/benchmarks/test_benchmark_mps_base.py -------------------------------------------------------------------------------- /torchmps/tests/benchmarks/test_benchmark_prob_mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/benchmarks/test_benchmark_prob_mps.py -------------------------------------------------------------------------------- /torchmps/tests/old_tests/custom_feature_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/old_tests/custom_feature_map.py -------------------------------------------------------------------------------- /torchmps/tests/old_tests/dynamic_mps_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/old_tests/dynamic_mps_basic.py -------------------------------------------------------------------------------- /torchmps/tests/old_tests/mat_region_open_periodic_bcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/old_tests/mat_region_open_periodic_bcs.py -------------------------------------------------------------------------------- /torchmps/tests/old_tests/static_mps_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/old_tests/static_mps_basic.py -------------------------------------------------------------------------------- /torchmps/tests/old_tests/svd_flex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/old_tests/svd_flex.py -------------------------------------------------------------------------------- /torchmps/tests/old_tests/ti_mps_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/old_tests/ti_mps_basic.py -------------------------------------------------------------------------------- /torchmps/tests/test_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/test_embeddings.py -------------------------------------------------------------------------------- /torchmps/tests/test_mps_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/test_mps_base.py -------------------------------------------------------------------------------- /torchmps/tests/test_prob_mps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/test_prob_mps.py -------------------------------------------------------------------------------- /torchmps/tests/test_utils2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/test_utils2.py -------------------------------------------------------------------------------- /torchmps/tests/utils_for_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/tests/utils_for_tests.py -------------------------------------------------------------------------------- /torchmps/torchmps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/torchmps.py -------------------------------------------------------------------------------- /torchmps/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/utils.py -------------------------------------------------------------------------------- /torchmps/utils2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/torchmps/utils2.py -------------------------------------------------------------------------------- /train_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jemisjoky/TorchMPS/HEAD/train_script.py --------------------------------------------------------------------------------