LICENSE
MANIFEST.in
NOTICE
README.md
pylintrc
pyproject.toml
pytest.ini
setup.py
example/run_bm.py
example/run_rbm.py
example/bm_generation/data_loader.py
example/bm_generation/sample_bm.ipynb
example/bm_generation/saver.py
example/bm_generation/train_bm.ipynb
example/bm_generation/trainer.py
example/dbn_digits/dbn_trainer.py
example/dbn_digits/supervised_dbn_digits.ipynb
example/dbn_digits/supervised_dbn_digits.py
example/qdiffusion/dplm/_example_bootstrap.py
example/qdiffusion/dplm/dplm_builder.py
example/qdiffusion/dplm/dplm_modeling.py
example/qdiffusion/dplm/eval_esm2_distances.py
example/qdiffusion/dplm/rerun_from_checkpoint.py
example/qdiffusion/dplm/train_workflow.py
example/qdiffusion/simple/_example_bootstrap.py
example/qdiffusion/simple/simple_generate_example.py
example/qdiffusion/simple/simple_train_example.py
example/qvae_cell/evaluate_benchmark.py
example/qvae_cell/models.py
example/qvae_cell/train_qvae_cell.ipynb
example/qvae_cell/train_qvae_cell.py
example/qvae_cell/trainer.py
example/qvae_cell/visualization.py
example/qvae_mnist/models.py
example/qvae_mnist/train_qvae.ipynb
example/qvae_mnist/train_qvae.py
example/qvae_mnist/train_qvae_classifier.ipynb
example/qvae_mnist/utils.py
example/qvae_mnist/visualizers.py
example/rbm_digits/rbm_classifier.py
example/rbm_digits/rbm_digits.ipynb
example/rbm_digits/rbm_digits.py
requirements/devel.txt
requirements/requirements.txt
src/kaiwu/torch_plugin/__init__.py
src/kaiwu/torch_plugin/_qdiffusion_sampling.py
src/kaiwu/torch_plugin/abstract_boltzmann_machine.py
src/kaiwu/torch_plugin/dbn.py
src/kaiwu/torch_plugin/full_boltzmann_machine.py
src/kaiwu/torch_plugin/qdiffusion.py
src/kaiwu/torch_plugin/qgan.py
src/kaiwu/torch_plugin/qvae.py
src/kaiwu/torch_plugin/qvae_dist_util.py
src/kaiwu/torch_plugin/restricted_boltzmann_machine.py
src/kaiwu_torch_plugin.egg-info/PKG-INFO
src/kaiwu_torch_plugin.egg-info/SOURCES.txt
src/kaiwu_torch_plugin.egg-info/dependency_links.txt
src/kaiwu_torch_plugin.egg-info/not-zip-safe
src/kaiwu_torch_plugin.egg-info/requires.txt
src/kaiwu_torch_plugin.egg-info/top_level.txt
tests/test_bm.py
tests/test_qdiffusion_dummy.py
tests/test_qdiffusion_import.py
tests/test_qvae.py
tests/test_rbm.py