├── .flake8 ├── .github ├── CONTRIBUTING.md ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── ci.yml │ ├── publish.yml │ └── test_tutorials.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── LICENSE ├── README.md ├── docs ├── Makefile ├── make.bat ├── requirements_docs.txt └── source │ ├── README.md │ ├── conf.py │ ├── guides │ ├── advanced.md │ ├── creating_environments.md │ ├── estimator_policy_mixin.md │ ├── example.md │ ├── losses.md │ ├── modules_estimators_samplers.md │ ├── states_actions_containers.md │ └── tutorials.md │ ├── index.rst │ └── logo.png ├── pyproject.toml ├── src └── gfn │ ├── __init__.py │ ├── actions.py │ ├── containers │ ├── __init__.py │ ├── base.py │ ├── message.py │ ├── replay_buffer.py │ ├── replay_buffer_manager.py │ ├── states_container.py │ ├── trajectories.py │ └── transitions.py │ ├── env.py │ ├── estimators.py │ ├── gflownet │ ├── __init__.py │ ├── base.py │ ├── detailed_balance.py │ ├── flow_matching.py │ ├── sub_trajectory_balance.py │ └── trajectory_balance.py │ ├── gym │ ├── __init__.py │ ├── bayesian_structure.py │ ├── bitSequence.py │ ├── box.py │ ├── diffusion_sampling.py │ ├── discrete_ebm.py │ ├── graph_building.py │ ├── helpers │ │ ├── __init__.py │ │ ├── bayesian_structure │ │ │ ├── __init__.py │ │ │ ├── evaluation.py │ │ │ ├── factories.py │ │ │ ├── graph.py │ │ │ ├── jsd.py │ │ │ ├── priors.py │ │ │ ├── sampling.py │ │ │ └── scores.py │ │ ├── box_utils.py │ │ └── diffusion_utils.py │ ├── hypergrid.py │ ├── line.py │ ├── perfect_tree.py │ ├── set_addition.py │ └── tests │ │ └── test_box_utils.py │ ├── modules.py │ ├── preprocessors.py │ ├── samplers.py │ ├── states.py │ └── utils │ ├── __init__.py │ ├── common.py │ ├── distributed.py │ ├── distributions.py │ ├── graphs.py │ ├── handlers.py │ ├── modules.py │ ├── prob_calculations.py │ └── training.py ├── testing ├── conftest.py ├── gym │ └── test_compile.py ├── test_actions.py ├── test_adaptor_estimator_gflownet_integration.py ├── test_containers.py ├── test_environments.py ├── test_estimators.py ├── test_gflownet.py ├── test_modules.py ├── test_parametrizations_and_losses.py ├── test_probability_calculations.py ├── test_replay_buffer.py ├── test_samplers_and_containers.py └── test_states.py └── tutorials ├── __init__.py ├── examples ├── README.md ├── __init__.py ├── mRNA_design │ ├── CAI_calculator.py │ ├── MFE_calculator.py │ ├── MFE_utils.py │ ├── README.md │ ├── config.yaml │ ├── env.py │ ├── evaluate.py │ ├── main.py │ ├── preprocessor.py │ ├── requirements.txt │ └── utils.py ├── multinode │ ├── .gitignore │ ├── README.md │ ├── example.ddp_gfn.slurm │ ├── install_multinode_dependencies │ └── spawn_policy.py ├── test_scripts.py ├── train_bayesian_structure.py ├── train_bit_sequences.py ├── train_bitsequence_recurrent.py ├── train_box.py ├── train_conditional.py ├── train_diffusion_sampler.py ├── train_discreteebm.py ├── train_graph_ring.py ├── train_graph_triangle.py ├── train_hypergrid.py ├── train_hypergrid_buffer.py ├── train_hypergrid_exploration_examples.py ├── train_hypergrid_gafn.py ├── train_hypergrid_local_search.py ├── train_hypergrid_simple.py ├── train_ising.py ├── train_line.py └── train_with_example_modes.py └── notebooks ├── README.md ├── intro_continuous.ipynb ├── intro_discrete.ipynb └── intro_graphs.ipynb /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.github/workflows/test_tutorials.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.github/workflows/test_tutorials.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/README.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements_docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/requirements_docs.txt -------------------------------------------------------------------------------- /docs/source/README.md: -------------------------------------------------------------------------------- 1 | ../../README.md -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/guides/advanced.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/advanced.md -------------------------------------------------------------------------------- /docs/source/guides/creating_environments.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/creating_environments.md -------------------------------------------------------------------------------- /docs/source/guides/estimator_policy_mixin.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/estimator_policy_mixin.md -------------------------------------------------------------------------------- /docs/source/guides/example.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/example.md -------------------------------------------------------------------------------- /docs/source/guides/losses.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/losses.md -------------------------------------------------------------------------------- /docs/source/guides/modules_estimators_samplers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/modules_estimators_samplers.md -------------------------------------------------------------------------------- /docs/source/guides/states_actions_containers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/states_actions_containers.md -------------------------------------------------------------------------------- /docs/source/guides/tutorials.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/guides/tutorials.md -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/docs/source/logo.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/gfn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/__init__.py -------------------------------------------------------------------------------- /src/gfn/actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/actions.py -------------------------------------------------------------------------------- /src/gfn/containers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/__init__.py -------------------------------------------------------------------------------- /src/gfn/containers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/base.py -------------------------------------------------------------------------------- /src/gfn/containers/message.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/message.py -------------------------------------------------------------------------------- /src/gfn/containers/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/replay_buffer.py -------------------------------------------------------------------------------- /src/gfn/containers/replay_buffer_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/replay_buffer_manager.py -------------------------------------------------------------------------------- /src/gfn/containers/states_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/states_container.py -------------------------------------------------------------------------------- /src/gfn/containers/trajectories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/trajectories.py -------------------------------------------------------------------------------- /src/gfn/containers/transitions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/containers/transitions.py -------------------------------------------------------------------------------- /src/gfn/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/env.py -------------------------------------------------------------------------------- /src/gfn/estimators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/estimators.py -------------------------------------------------------------------------------- /src/gfn/gflownet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gflownet/__init__.py -------------------------------------------------------------------------------- /src/gfn/gflownet/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gflownet/base.py -------------------------------------------------------------------------------- /src/gfn/gflownet/detailed_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gflownet/detailed_balance.py -------------------------------------------------------------------------------- /src/gfn/gflownet/flow_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gflownet/flow_matching.py -------------------------------------------------------------------------------- /src/gfn/gflownet/sub_trajectory_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gflownet/sub_trajectory_balance.py -------------------------------------------------------------------------------- /src/gfn/gflownet/trajectory_balance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gflownet/trajectory_balance.py -------------------------------------------------------------------------------- /src/gfn/gym/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/__init__.py -------------------------------------------------------------------------------- /src/gfn/gym/bayesian_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/bayesian_structure.py -------------------------------------------------------------------------------- /src/gfn/gym/bitSequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/bitSequence.py -------------------------------------------------------------------------------- /src/gfn/gym/box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/box.py -------------------------------------------------------------------------------- /src/gfn/gym/diffusion_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/diffusion_sampling.py -------------------------------------------------------------------------------- /src/gfn/gym/discrete_ebm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/discrete_ebm.py -------------------------------------------------------------------------------- /src/gfn/gym/graph_building.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/graph_building.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/__init__.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/evaluation.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/factories.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/factories.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/graph.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/jsd.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/priors.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/sampling.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/bayesian_structure/scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/bayesian_structure/scores.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/box_utils.py -------------------------------------------------------------------------------- /src/gfn/gym/helpers/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/helpers/diffusion_utils.py -------------------------------------------------------------------------------- /src/gfn/gym/hypergrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/hypergrid.py -------------------------------------------------------------------------------- /src/gfn/gym/line.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/line.py -------------------------------------------------------------------------------- /src/gfn/gym/perfect_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/perfect_tree.py -------------------------------------------------------------------------------- /src/gfn/gym/set_addition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/set_addition.py -------------------------------------------------------------------------------- /src/gfn/gym/tests/test_box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/gym/tests/test_box_utils.py -------------------------------------------------------------------------------- /src/gfn/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/modules.py -------------------------------------------------------------------------------- /src/gfn/preprocessors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/preprocessors.py -------------------------------------------------------------------------------- /src/gfn/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/samplers.py -------------------------------------------------------------------------------- /src/gfn/states.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/states.py -------------------------------------------------------------------------------- /src/gfn/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/gfn/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/common.py -------------------------------------------------------------------------------- /src/gfn/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/distributed.py -------------------------------------------------------------------------------- /src/gfn/utils/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/distributions.py -------------------------------------------------------------------------------- /src/gfn/utils/graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/graphs.py -------------------------------------------------------------------------------- /src/gfn/utils/handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/handlers.py -------------------------------------------------------------------------------- /src/gfn/utils/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/modules.py -------------------------------------------------------------------------------- /src/gfn/utils/prob_calculations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/prob_calculations.py -------------------------------------------------------------------------------- /src/gfn/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/src/gfn/utils/training.py -------------------------------------------------------------------------------- /testing/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/conftest.py -------------------------------------------------------------------------------- /testing/gym/test_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/gym/test_compile.py -------------------------------------------------------------------------------- /testing/test_actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_actions.py -------------------------------------------------------------------------------- /testing/test_adaptor_estimator_gflownet_integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_adaptor_estimator_gflownet_integration.py -------------------------------------------------------------------------------- /testing/test_containers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_containers.py -------------------------------------------------------------------------------- /testing/test_environments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_environments.py -------------------------------------------------------------------------------- /testing/test_estimators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_estimators.py -------------------------------------------------------------------------------- /testing/test_gflownet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_gflownet.py -------------------------------------------------------------------------------- /testing/test_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_modules.py -------------------------------------------------------------------------------- /testing/test_parametrizations_and_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_parametrizations_and_losses.py -------------------------------------------------------------------------------- /testing/test_probability_calculations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_probability_calculations.py -------------------------------------------------------------------------------- /testing/test_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_replay_buffer.py -------------------------------------------------------------------------------- /testing/test_samplers_and_containers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_samplers_and_containers.py -------------------------------------------------------------------------------- /testing/test_states.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/testing/test_states.py -------------------------------------------------------------------------------- /tutorials/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tutorials/examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/README.md -------------------------------------------------------------------------------- /tutorials/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/CAI_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/CAI_calculator.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/MFE_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/MFE_calculator.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/MFE_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/MFE_utils.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/README.md -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/config.yaml -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/env.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/evaluate.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/main.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/preprocessor.py -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/requirements.txt: -------------------------------------------------------------------------------- 1 | CAI 2 | cai-pypi==2.0.1 3 | -------------------------------------------------------------------------------- /tutorials/examples/mRNA_design/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/mRNA_design/utils.py -------------------------------------------------------------------------------- /tutorials/examples/multinode/.gitignore: -------------------------------------------------------------------------------- 1 | logs 2 | -------------------------------------------------------------------------------- /tutorials/examples/multinode/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/multinode/README.md -------------------------------------------------------------------------------- /tutorials/examples/multinode/example.ddp_gfn.slurm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/multinode/example.ddp_gfn.slurm -------------------------------------------------------------------------------- /tutorials/examples/multinode/install_multinode_dependencies: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/multinode/install_multinode_dependencies -------------------------------------------------------------------------------- /tutorials/examples/multinode/spawn_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/multinode/spawn_policy.py -------------------------------------------------------------------------------- /tutorials/examples/test_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/test_scripts.py -------------------------------------------------------------------------------- /tutorials/examples/train_bayesian_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_bayesian_structure.py -------------------------------------------------------------------------------- /tutorials/examples/train_bit_sequences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_bit_sequences.py -------------------------------------------------------------------------------- /tutorials/examples/train_bitsequence_recurrent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_bitsequence_recurrent.py -------------------------------------------------------------------------------- /tutorials/examples/train_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_box.py -------------------------------------------------------------------------------- /tutorials/examples/train_conditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_conditional.py -------------------------------------------------------------------------------- /tutorials/examples/train_diffusion_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_diffusion_sampler.py -------------------------------------------------------------------------------- /tutorials/examples/train_discreteebm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_discreteebm.py -------------------------------------------------------------------------------- /tutorials/examples/train_graph_ring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_graph_ring.py -------------------------------------------------------------------------------- /tutorials/examples/train_graph_triangle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_graph_triangle.py -------------------------------------------------------------------------------- /tutorials/examples/train_hypergrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_hypergrid.py -------------------------------------------------------------------------------- /tutorials/examples/train_hypergrid_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_hypergrid_buffer.py -------------------------------------------------------------------------------- /tutorials/examples/train_hypergrid_exploration_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_hypergrid_exploration_examples.py -------------------------------------------------------------------------------- /tutorials/examples/train_hypergrid_gafn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_hypergrid_gafn.py -------------------------------------------------------------------------------- /tutorials/examples/train_hypergrid_local_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_hypergrid_local_search.py -------------------------------------------------------------------------------- /tutorials/examples/train_hypergrid_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_hypergrid_simple.py -------------------------------------------------------------------------------- /tutorials/examples/train_ising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_ising.py -------------------------------------------------------------------------------- /tutorials/examples/train_line.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_line.py -------------------------------------------------------------------------------- /tutorials/examples/train_with_example_modes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/examples/train_with_example_modes.py -------------------------------------------------------------------------------- /tutorials/notebooks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/notebooks/README.md -------------------------------------------------------------------------------- /tutorials/notebooks/intro_continuous.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/notebooks/intro_continuous.ipynb -------------------------------------------------------------------------------- /tutorials/notebooks/intro_discrete.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/notebooks/intro_discrete.ipynb -------------------------------------------------------------------------------- /tutorials/notebooks/intro_graphs.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/torchgfn/HEAD/tutorials/notebooks/intro_graphs.ipynb --------------------------------------------------------------------------------