├── .github └── workflows │ ├── pythonpackage.yml │ └── pythonpublish.yml ├── .gitignore ├── .travis.yml ├── LICENSE ├── README.md ├── examples ├── basic_autoregressive_2d.py ├── basic_autoregressive_exact_gradient.py ├── basic_autoregressive_exact_gradient_multy_gpu.py ├── basic_autoregressive_monte_carlo_gradient.py ├── benchmark.py ├── benchmark_sampling.py ├── complex_ops_autoregressive_heisenberg_1d.py ├── custom_keras_model.py ├── custom_keras_model_with_validation_data.py ├── debug_fast_sampling.py ├── evaluate_with_2d_obc_invariant.py ├── heisenberg_2d_horvod_multy_gpu_fast_sampling.py ├── heisenberg_2d_keras_multy_gpu_fast_sampling.py ├── ising.py ├── j1j2_2d_exact_4.py ├── j1j2_2d_monte_carlo_4.py └── rbm_heisenberg_1d_sr.py ├── experiments ├── README.md ├── heisenberg_runner.py ├── ising_runner.py ├── run_evaluation.py ├── train.py └── weights │ ├── heisenberg.h5 │ ├── ising_2.h5 │ ├── ising_2_5.h5 │ ├── ising_3.h5 │ ├── ising_3_5.h5 │ └── ising_4.h5 ├── setup.cfg ├── setup.py ├── src └── flowket │ ├── __init__.py │ ├── callbacks │ ├── __init__.py │ ├── checkpoint.py │ ├── exact │ │ ├── __init__.py │ │ ├── local_energy.py │ │ ├── machine_updated.py │ │ ├── observable.py │ │ ├── runtime_stats.py │ │ └── sigma_z.py │ ├── monte_carlo │ │ ├── __init__.py │ │ ├── bad_eigen_state_stopping.py │ │ ├── generator_iterator.py │ │ ├── local_energy_stats.py │ │ ├── mcmc_stats.py │ │ ├── observable.py │ │ ├── runtime_stats.py │ │ └── tensorboard_with_generator_validation_data.py │ └── tensorboard.py │ ├── deepar │ ├── README.md │ ├── __init__.py │ ├── graph_analysis │ │ ├── __init__.py │ │ ├── concatenate_topology.py │ │ ├── convolutional_topology.py │ │ ├── data_structures.py │ │ ├── dependency_graph.py │ │ ├── gathering_topology.py │ │ ├── layer_topology.py │ │ ├── masking_topology.py │ │ ├── one_to_one_topology.py │ │ ├── padding_topology.py │ │ ├── reshape_topology.py │ │ ├── sampling_topology.py │ │ └── topology_manager.py │ ├── layers │ │ ├── __init__.py │ │ ├── autoregressive.py │ │ ├── casting.py │ │ ├── gathering.py │ │ ├── lambda_with_one_to_one_topology.py │ │ ├── layer_normalization.py │ │ ├── masking.py │ │ ├── one_hot.py │ │ ├── padding.py │ │ └── wrappers.py │ ├── ordering │ │ ├── __init__.py │ │ ├── flat.py │ │ ├── moves.py │ │ ├── raster.py │ │ └── zigzag.py │ ├── samplers │ │ ├── __init__.py │ │ ├── autoregressive.py │ │ ├── base_sampler.py │ │ ├── ensemble.py │ │ └── fast_autoregressive.py │ └── utils │ │ ├── __init__.py │ │ └── singleton.py │ ├── evaluation │ ├── __init__.py │ └── evaluate.py │ ├── exact │ ├── __init__.py │ └── utils.py │ ├── layers │ ├── __init__.py │ ├── complex │ │ ├── __init__.py │ │ ├── base_layer.py │ │ ├── casting.py │ │ ├── conv.py │ │ ├── dense.py │ │ ├── histograms.py │ │ ├── initializers.py │ │ └── tensorflow_ops.py │ ├── dihedral_4_invariants.py │ ├── spins_invariants.py │ └── transition_invariants.py │ ├── machines │ ├── __init__.py │ ├── abstract_machine.py │ ├── complex_values_simple_conv_net_autoregressive_1D.py │ ├── conv_net_autoregressive_2D.py │ ├── ensemble.py │ ├── rbm.py │ ├── simple_conv_net_autoregressive_1D.py │ └── simple_custom_autoregressive_ordering.py │ ├── main.py │ ├── observables │ ├── __init__.py │ └── monte_carlo │ │ ├── __init__.py │ │ ├── observable.py │ │ ├── operator.py │ │ └── sigma_z.py │ ├── operators │ ├── __init__.py │ ├── heisenberg.py │ ├── ising.py │ ├── j1j2.py │ ├── netket_operator.py │ └── operator.py │ ├── optimization │ ├── __init__.py │ ├── exact_variational.py │ ├── horovod_variational_monte_carlo.py │ ├── loss.py │ ├── mini_batch_generator.py │ └── variational_monte_carlo.py │ ├── optimizers │ ├── __init__.py │ ├── accumulate_gradient_optimizer.py │ ├── complex_values_optimizer.py │ ├── stochastic_reconfiguration │ │ ├── __init__.py │ │ ├── linear_equations.py │ │ └── optimizer.py │ └── utils.py │ ├── samplers │ ├── __init__.py │ ├── exact_sampler.py │ ├── fast_autoregressive │ │ └── __init__.py │ └── metropolis_hastings.py │ └── utils │ ├── __init__.py │ ├── jacobian.py │ └── v1_to_v2.py └── tests ├── __init__.py ├── conftest.py ├── simple_models.py ├── test_autoregressive.py ├── test_complex_values_optimizer.py ├── test_ensemble.py ├── test_gradient_per_example.py ├── test_gradients_aggregation.py ├── test_graph_analysis.py ├── test_samplers.py ├── test_stochastic_reconfiguration.py ├── test_tensorflow_complex_numbers_ops.py └── test_variational.py /.github/workflows/pythonpackage.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/.github/workflows/pythonpackage.yml -------------------------------------------------------------------------------- /.github/workflows/pythonpublish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/.github/workflows/pythonpublish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/README.md -------------------------------------------------------------------------------- /examples/basic_autoregressive_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/basic_autoregressive_2d.py -------------------------------------------------------------------------------- /examples/basic_autoregressive_exact_gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/basic_autoregressive_exact_gradient.py -------------------------------------------------------------------------------- /examples/basic_autoregressive_exact_gradient_multy_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/basic_autoregressive_exact_gradient_multy_gpu.py -------------------------------------------------------------------------------- /examples/basic_autoregressive_monte_carlo_gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/basic_autoregressive_monte_carlo_gradient.py -------------------------------------------------------------------------------- /examples/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/benchmark.py -------------------------------------------------------------------------------- /examples/benchmark_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/benchmark_sampling.py -------------------------------------------------------------------------------- /examples/complex_ops_autoregressive_heisenberg_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/complex_ops_autoregressive_heisenberg_1d.py -------------------------------------------------------------------------------- /examples/custom_keras_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/custom_keras_model.py -------------------------------------------------------------------------------- /examples/custom_keras_model_with_validation_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/custom_keras_model_with_validation_data.py -------------------------------------------------------------------------------- /examples/debug_fast_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/debug_fast_sampling.py -------------------------------------------------------------------------------- /examples/evaluate_with_2d_obc_invariant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/evaluate_with_2d_obc_invariant.py -------------------------------------------------------------------------------- /examples/heisenberg_2d_horvod_multy_gpu_fast_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/heisenberg_2d_horvod_multy_gpu_fast_sampling.py -------------------------------------------------------------------------------- /examples/heisenberg_2d_keras_multy_gpu_fast_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/heisenberg_2d_keras_multy_gpu_fast_sampling.py -------------------------------------------------------------------------------- /examples/ising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/ising.py -------------------------------------------------------------------------------- /examples/j1j2_2d_exact_4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/j1j2_2d_exact_4.py -------------------------------------------------------------------------------- /examples/j1j2_2d_monte_carlo_4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/j1j2_2d_monte_carlo_4.py -------------------------------------------------------------------------------- /examples/rbm_heisenberg_1d_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/examples/rbm_heisenberg_1d_sr.py -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/README.md -------------------------------------------------------------------------------- /experiments/heisenberg_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/heisenberg_runner.py -------------------------------------------------------------------------------- /experiments/ising_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/ising_runner.py -------------------------------------------------------------------------------- /experiments/run_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/run_evaluation.py -------------------------------------------------------------------------------- /experiments/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/train.py -------------------------------------------------------------------------------- /experiments/weights/heisenberg.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/weights/heisenberg.h5 -------------------------------------------------------------------------------- /experiments/weights/ising_2.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/weights/ising_2.h5 -------------------------------------------------------------------------------- /experiments/weights/ising_2_5.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/weights/ising_2_5.h5 -------------------------------------------------------------------------------- /experiments/weights/ising_3.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/weights/ising_3.h5 -------------------------------------------------------------------------------- /experiments/weights/ising_3_5.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/weights/ising_3_5.h5 -------------------------------------------------------------------------------- /experiments/weights/ising_4.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/experiments/weights/ising_4.h5 -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/setup.py -------------------------------------------------------------------------------- /src/flowket/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/__init__.py -------------------------------------------------------------------------------- /src/flowket/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/__init__.py -------------------------------------------------------------------------------- /src/flowket/callbacks/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/checkpoint.py -------------------------------------------------------------------------------- /src/flowket/callbacks/exact/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/exact/__init__.py -------------------------------------------------------------------------------- /src/flowket/callbacks/exact/local_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/exact/local_energy.py -------------------------------------------------------------------------------- /src/flowket/callbacks/exact/machine_updated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/exact/machine_updated.py -------------------------------------------------------------------------------- /src/flowket/callbacks/exact/observable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/exact/observable.py -------------------------------------------------------------------------------- /src/flowket/callbacks/exact/runtime_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/exact/runtime_stats.py -------------------------------------------------------------------------------- /src/flowket/callbacks/exact/sigma_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/exact/sigma_z.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/__init__.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/bad_eigen_state_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/bad_eigen_state_stopping.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/generator_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/generator_iterator.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/local_energy_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/local_energy_stats.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/mcmc_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/mcmc_stats.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/observable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/observable.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/runtime_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/runtime_stats.py -------------------------------------------------------------------------------- /src/flowket/callbacks/monte_carlo/tensorboard_with_generator_validation_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/monte_carlo/tensorboard_with_generator_validation_data.py -------------------------------------------------------------------------------- /src/flowket/callbacks/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/callbacks/tensorboard.py -------------------------------------------------------------------------------- /src/flowket/deepar/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/README.md -------------------------------------------------------------------------------- /src/flowket/deepar/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/__init__.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/concatenate_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/concatenate_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/convolutional_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/convolutional_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/data_structures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/data_structures.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/dependency_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/dependency_graph.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/gathering_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/gathering_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/layer_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/layer_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/masking_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/masking_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/one_to_one_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/one_to_one_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/padding_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/padding_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/reshape_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/reshape_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/sampling_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/sampling_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/graph_analysis/topology_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/graph_analysis/topology_manager.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/__init__.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/autoregressive.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/casting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/casting.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/gathering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/gathering.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/lambda_with_one_to_one_topology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/lambda_with_one_to_one_topology.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/layer_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/layer_normalization.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/masking.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/one_hot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/one_hot.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/padding.py -------------------------------------------------------------------------------- /src/flowket/deepar/layers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/layers/wrappers.py -------------------------------------------------------------------------------- /src/flowket/deepar/ordering/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/ordering/__init__.py -------------------------------------------------------------------------------- /src/flowket/deepar/ordering/flat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/ordering/flat.py -------------------------------------------------------------------------------- /src/flowket/deepar/ordering/moves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/ordering/moves.py -------------------------------------------------------------------------------- /src/flowket/deepar/ordering/raster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/ordering/raster.py -------------------------------------------------------------------------------- /src/flowket/deepar/ordering/zigzag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/ordering/zigzag.py -------------------------------------------------------------------------------- /src/flowket/deepar/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/samplers/__init__.py -------------------------------------------------------------------------------- /src/flowket/deepar/samplers/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/samplers/autoregressive.py -------------------------------------------------------------------------------- /src/flowket/deepar/samplers/base_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/samplers/base_sampler.py -------------------------------------------------------------------------------- /src/flowket/deepar/samplers/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/samplers/ensemble.py -------------------------------------------------------------------------------- /src/flowket/deepar/samplers/fast_autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/samplers/fast_autoregressive.py -------------------------------------------------------------------------------- /src/flowket/deepar/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/utils/__init__.py -------------------------------------------------------------------------------- /src/flowket/deepar/utils/singleton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/deepar/utils/singleton.py -------------------------------------------------------------------------------- /src/flowket/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/evaluation/__init__.py -------------------------------------------------------------------------------- /src/flowket/evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/evaluation/evaluate.py -------------------------------------------------------------------------------- /src/flowket/exact/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/flowket/exact/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/exact/utils.py -------------------------------------------------------------------------------- /src/flowket/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/__init__.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/flowket/layers/complex/base_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/base_layer.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/casting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/casting.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/conv.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/dense.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/histograms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/histograms.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/initializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/initializers.py -------------------------------------------------------------------------------- /src/flowket/layers/complex/tensorflow_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/complex/tensorflow_ops.py -------------------------------------------------------------------------------- /src/flowket/layers/dihedral_4_invariants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/dihedral_4_invariants.py -------------------------------------------------------------------------------- /src/flowket/layers/spins_invariants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/spins_invariants.py -------------------------------------------------------------------------------- /src/flowket/layers/transition_invariants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/layers/transition_invariants.py -------------------------------------------------------------------------------- /src/flowket/machines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/__init__.py -------------------------------------------------------------------------------- /src/flowket/machines/abstract_machine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/abstract_machine.py -------------------------------------------------------------------------------- /src/flowket/machines/complex_values_simple_conv_net_autoregressive_1D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/complex_values_simple_conv_net_autoregressive_1D.py -------------------------------------------------------------------------------- /src/flowket/machines/conv_net_autoregressive_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/conv_net_autoregressive_2D.py -------------------------------------------------------------------------------- /src/flowket/machines/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/ensemble.py -------------------------------------------------------------------------------- /src/flowket/machines/rbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/rbm.py -------------------------------------------------------------------------------- /src/flowket/machines/simple_conv_net_autoregressive_1D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/simple_conv_net_autoregressive_1D.py -------------------------------------------------------------------------------- /src/flowket/machines/simple_custom_autoregressive_ordering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/machines/simple_custom_autoregressive_ordering.py -------------------------------------------------------------------------------- /src/flowket/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/main.py -------------------------------------------------------------------------------- /src/flowket/observables/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/flowket/observables/monte_carlo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/observables/monte_carlo/__init__.py -------------------------------------------------------------------------------- /src/flowket/observables/monte_carlo/observable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/observables/monte_carlo/observable.py -------------------------------------------------------------------------------- /src/flowket/observables/monte_carlo/operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/observables/monte_carlo/operator.py -------------------------------------------------------------------------------- /src/flowket/observables/monte_carlo/sigma_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/observables/monte_carlo/sigma_z.py -------------------------------------------------------------------------------- /src/flowket/operators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/operators/__init__.py -------------------------------------------------------------------------------- /src/flowket/operators/heisenberg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/operators/heisenberg.py -------------------------------------------------------------------------------- /src/flowket/operators/ising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/operators/ising.py -------------------------------------------------------------------------------- /src/flowket/operators/j1j2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/operators/j1j2.py -------------------------------------------------------------------------------- /src/flowket/operators/netket_operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/operators/netket_operator.py -------------------------------------------------------------------------------- /src/flowket/operators/operator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/operators/operator.py -------------------------------------------------------------------------------- /src/flowket/optimization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimization/__init__.py -------------------------------------------------------------------------------- /src/flowket/optimization/exact_variational.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimization/exact_variational.py -------------------------------------------------------------------------------- /src/flowket/optimization/horovod_variational_monte_carlo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimization/horovod_variational_monte_carlo.py -------------------------------------------------------------------------------- /src/flowket/optimization/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimization/loss.py -------------------------------------------------------------------------------- /src/flowket/optimization/mini_batch_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimization/mini_batch_generator.py -------------------------------------------------------------------------------- /src/flowket/optimization/variational_monte_carlo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimization/variational_monte_carlo.py -------------------------------------------------------------------------------- /src/flowket/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimizers/__init__.py -------------------------------------------------------------------------------- /src/flowket/optimizers/accumulate_gradient_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimizers/accumulate_gradient_optimizer.py -------------------------------------------------------------------------------- /src/flowket/optimizers/complex_values_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimizers/complex_values_optimizer.py -------------------------------------------------------------------------------- /src/flowket/optimizers/stochastic_reconfiguration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/flowket/optimizers/stochastic_reconfiguration/linear_equations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimizers/stochastic_reconfiguration/linear_equations.py -------------------------------------------------------------------------------- /src/flowket/optimizers/stochastic_reconfiguration/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimizers/stochastic_reconfiguration/optimizer.py -------------------------------------------------------------------------------- /src/flowket/optimizers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/optimizers/utils.py -------------------------------------------------------------------------------- /src/flowket/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/samplers/__init__.py -------------------------------------------------------------------------------- /src/flowket/samplers/exact_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/samplers/exact_sampler.py -------------------------------------------------------------------------------- /src/flowket/samplers/fast_autoregressive/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/samplers/fast_autoregressive/__init__.py -------------------------------------------------------------------------------- /src/flowket/samplers/metropolis_hastings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/samplers/metropolis_hastings.py -------------------------------------------------------------------------------- /src/flowket/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/flowket/utils/jacobian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/utils/jacobian.py -------------------------------------------------------------------------------- /src/flowket/utils/v1_to_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/src/flowket/utils/v1_to_v2.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/simple_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/simple_models.py -------------------------------------------------------------------------------- /tests/test_autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_autoregressive.py -------------------------------------------------------------------------------- /tests/test_complex_values_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_complex_values_optimizer.py -------------------------------------------------------------------------------- /tests/test_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_ensemble.py -------------------------------------------------------------------------------- /tests/test_gradient_per_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_gradient_per_example.py -------------------------------------------------------------------------------- /tests/test_gradients_aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_gradients_aggregation.py -------------------------------------------------------------------------------- /tests/test_graph_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_graph_analysis.py -------------------------------------------------------------------------------- /tests/test_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_samplers.py -------------------------------------------------------------------------------- /tests/test_stochastic_reconfiguration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_stochastic_reconfiguration.py -------------------------------------------------------------------------------- /tests/test_tensorflow_complex_numbers_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_tensorflow_complex_numbers_ops.py -------------------------------------------------------------------------------- /tests/test_variational.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HUJI-Deep/FlowKet/HEAD/tests/test_variational.py --------------------------------------------------------------------------------