Index

_

__call__() (gfn.preprocessors.Preprocessor method) __getitem__() (gfn.actions.Actions method) (gfn.containers.base.Container method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.states.DiscreteStates method) (gfn.states.States method) __len__() (gfn.actions.Actions method) (gfn.containers.base.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.states.States method)

__repr__() (gfn.actions.Actions method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.modules.GFNModule method) (gfn.preprocessors.Preprocessor method) (gfn.states.States method) __setitem__() (gfn.actions.Actions method) (gfn.states.DiscreteStates method) (gfn.states.States method) __version__ (in module gfn) _check_both_forward_backward_masks_exist() (gfn.states.DiscreteStates method) _log_rewards (gfn.states.States attribute) _output_dim_is_checked (gfn.modules.GFNModule attribute) (gfn.modules.ScalarEstimator attribute)

A

action_shape (gfn.actions.Actions attribute) Actions (class in gfn.actions) actions (gfn.containers.Trajectories attribute) (gfn.containers.trajectories.Trajectories attribute) (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute) add() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) all_log_rewards (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property)

all_states (gfn.env.DiscreteEnv property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) alpha (examples.test_scripts.DiscreteEBMArgs attribute) (gfn.gflownet.flow_matching.FMGFlowNet attribute) (gfn.gflownet.FMGFlowNet attribute)

B

backward_masks (gfn.states.DiscreteStates attribute) backward_step() (gfn.env.Env method) batch_shape (gfn.actions.Actions attribute) (gfn.states.States attribute) batch_size (examples.test_scripts.CommonArgs attribute) Box (class in gfn.gym) (class in gfn.gym.box) BoxArgs (class in examples.test_scripts)

BoxPBEstimator (class in gfn.gym.helpers.box_utils) BoxPBNeuralNet (class in gfn.gym.helpers.box_utils) BoxPBUniform (class in gfn.gym.helpers.box_utils) BoxPFEstimator (class in gfn.gym.helpers.box_utils) BoxPFNeuralNet (class in gfn.gym.helpers.box_utils) BoxStateFlowModule (class in gfn.gym.helpers.box_utils) build_grid() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method)

C

calculate_log_state_flows() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) calculate_masks() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) calculate_preds() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) calculate_targets() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) capacity (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute)

check_output_dim() (gfn.modules.GFNModule method) CLAMP (in module gfn.gym.helpers.box_utils) CommonArgs (class in examples.test_scripts) compare() (gfn.actions.Actions method) (gfn.states.States method) Container (class in gfn.containers.base) ContributionsTensor (in module gfn.gflownet.sub_trajectory_balance) cumulative_logprobs() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) CumulativeLogProbsTensor (in module gfn.gflownet.sub_trajectory_balance)

D

DBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.detailed_balance) delta (examples.test_scripts.BoxArgs attribute) device (gfn.actions.Actions property) (gfn.states.DiscreteStates attribute) (gfn.states.States property) (gfn.utils.modules.Tabular attribute) (gfn.utils.Tabular attribute)

DiscreteEBM (class in gfn.gym) (class in gfn.gym.discrete_ebm) DiscreteEBMArgs (class in examples.test_scripts) DiscreteEnv (class in gfn.env) DiscretePolicyEstimator (class in gfn.modules) DiscreteStates (class in gfn.states) DiscreteUniform (class in gfn.utils.modules) DistributionWrapper (class in gfn.gym.helpers.box_utils) dummy_action (gfn.actions.Actions attribute)

E

EnergyFunction (class in gfn.gym.discrete_ebm) EnumPreprocessor (class in gfn.preprocessors) Env (class in gfn.env) env (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) (gfn.containers.Trajectories attribute) (gfn.containers.trajectories.Trajectories attribute) (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute) epsilon (gfn.modules.DiscretePolicyEstimator attribute) estimate_jsd() (in module examples.train_box) estimator (gfn.samplers.Sampler attribute) examples module examples.test_scripts module examples.train_box module examples.train_discreteebm module

examples.train_hypergrid module exit_action (gfn.actions.Actions attribute) expected_output_dim (gfn.modules.GFNModule property) expected_output_dim() (gfn.gym.helpers.box_utils.BoxPBEstimator method) (gfn.gym.helpers.box_utils.BoxPFEstimator method) (gfn.modules.DiscretePolicyEstimator method) (gfn.modules.ScalarEstimator method) extend() (gfn.actions.Actions method) (gfn.containers.base.Container method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.states.DiscreteStates method) (gfn.states.States method) extend_log_probs() (gfn.containers.Trajectories static method) (gfn.containers.trajectories.Trajectories static method) extend_with_dummy_actions() (gfn.actions.Actions method) extend_with_sf() (gfn.states.DiscreteStates method) (gfn.states.States method)

F

flatten() (gfn.states.DiscreteStates method) (gfn.states.States method) flow_matching_loss() (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method) FMGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.flow_matching) forward() (gfn.gym.discrete_ebm.EnergyFunction method) (gfn.gym.discrete_ebm.IsingModel method) (gfn.gym.helpers.box_utils.BoxPBNeuralNet method) (gfn.gym.helpers.box_utils.BoxPBUniform method) (gfn.gym.helpers.box_utils.BoxPFNeuralNet method) (gfn.gym.helpers.box_utils.BoxStateFlowModule method) (gfn.modules.GFNModule method) (gfn.utils.modules.DiscreteUniform method) (gfn.utils.modules.NeuralNet method) (gfn.utils.modules.Tabular method) (gfn.utils.NeuralNet method) (gfn.utils.Tabular method)

forward_looking (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute) forward_masks (gfn.states.DiscreteStates attribute) from_batch_shape() (gfn.states.States class method)

G

gamma_scheduler (examples.test_scripts.BoxArgs attribute) get_equal_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_equal_within_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_geometric_within_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_min_and_max_angles() (gfn.gym.helpers.box_utils.QuarterCircle method) get_modified_db_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_pfs_and_pbs() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) get_scores() (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet method) (gfn.gflownet.ModifiedDBGFlowNet method) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_states_indices() (gfn.env.DiscreteEnv method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) get_tb_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_terminating_state_dist_pmf() (in module gfn.utils.common) get_terminating_states_indices() (gfn.env.DiscreteEnv method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) get_test_states() (in module examples.train_box) get_trajectories_scores() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) GFlowNet (class in gfn.gflownet) (class in gfn.gflownet.base) gfn module gfn.actions module gfn.containers module gfn.containers.base module gfn.containers.replay_buffer module gfn.containers.trajectories module

gfn.containers.transitions module gfn.env module gfn.gflownet module gfn.gflownet.base module gfn.gflownet.detailed_balance module gfn.gflownet.flow_matching module gfn.gflownet.sub_trajectory_balance module gfn.gflownet.trajectory_balance module gfn.gym module gfn.gym.box module gfn.gym.discrete_ebm module gfn.gym.helpers module gfn.gym.helpers.box_utils module gfn.gym.helpers.preprocessors module gfn.gym.helpers.test_box_utils module gfn.gym.hypergrid module gfn.modules module gfn.preprocessors module gfn.samplers module gfn.states module gfn.utils module gfn.utils.common module gfn.utils.distributions module gfn.utils.modules module GFNModule (class in gfn.modules)

H

height (examples.test_scripts.HypergridArgs attribute) hidden_dim (examples.test_scripts.CommonArgs attribute)

HyperGrid (class in gfn.gym) (class in gfn.gym.hypergrid) HypergridArgs (class in examples.test_scripts)

I

IdentityPreprocessor (class in gfn.preprocessors) input_dim (gfn.gym.helpers.box_utils.BoxPBUniform attribute) is_action_valid() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.gym.Box method) (gfn.gym.box.Box method) is_backward (gfn.containers.Trajectories attribute) (gfn.containers.trajectories.Trajectories attribute) (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute)

is_done (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute) is_dummy (gfn.actions.Actions property) is_exit (gfn.actions.Actions property) is_exit_actions() (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) is_initial_state (gfn.states.States property) is_sink_state (gfn.states.States property) IsingModel (class in gfn.gym.discrete_ebm)

K

KHotPreprocessor (class in gfn.gym.helpers.preprocessors)

L

lamda (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute) (gfn.gflownet.SubTBGFlowNet attribute) last_states (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) load() (gfn.containers.base.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) log_partition (gfn.env.Env property) (gfn.gym.Box property) (gfn.gym.box.Box property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) log_prob() (gfn.gym.helpers.box_utils.DistributionWrapper method) (gfn.gym.helpers.box_utils.QuarterCircle method) (gfn.gym.helpers.box_utils.QuarterCircleWithExit method) (gfn.gym.helpers.box_utils.QuarterDisk method) (gfn.utils.distributions.UnsqueezedCategorical method) log_probs (gfn.containers.Trajectories attribute) (gfn.containers.trajectories.Trajectories attribute) (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute) log_reward() (gfn.env.Env method) (gfn.gym.Box method) (gfn.gym.box.Box method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) log_reward_clip_min (gfn.gflownet.LogPartitionVarianceGFlowNet attribute) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute) (gfn.gflownet.SubTBGFlowNet attribute) (gfn.gflownet.TBGFlowNet attribute) (gfn.gflownet.trajectory_balance.LogPartitionVarianceGFlowNet attribute) (gfn.gflownet.trajectory_balance.TBGFlowNet attribute)

log_rewards (gfn.containers.Trajectories attribute) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories attribute) (gfn.containers.trajectories.Trajectories property) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) (gfn.states.States property) logF (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute) (gfn.gflownet.flow_matching.FMGFlowNet attribute) (gfn.gflownet.FMGFlowNet attribute) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute) (gfn.gflownet.SubTBGFlowNet attribute) LogPartitionVarianceGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) LogStateFlowsTensor (in module gfn.gflownet.sub_trajectory_balance) LogTrajectoriesTensor (in module gfn.gflownet.sub_trajectory_balance) logZ (gfn.gflownet.TBGFlowNet attribute) (gfn.gflownet.trajectory_balance.TBGFlowNet attribute) loss (examples.test_scripts.CommonArgs attribute) loss() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet method) (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method) (gfn.gflownet.GFlowNet method) (gfn.gflownet.LogPartitionVarianceGFlowNet method) (gfn.gflownet.ModifiedDBGFlowNet method) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) (gfn.gflownet.TBGFlowNet method) (gfn.gflownet.trajectory_balance.LogPartitionVarianceGFlowNet method) (gfn.gflownet.trajectory_balance.TBGFlowNet method) loss_fn (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) lr (examples.test_scripts.CommonArgs attribute) lr_F (examples.test_scripts.BoxArgs attribute) lr_Z (examples.test_scripts.CommonArgs attribute)

M

main() (in module examples.train_box) (in module examples.train_discreteebm) (in module examples.train_hypergrid) make_Actions_class() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.gym.Box method) (gfn.gym.box.Box method) make_dummy_actions() (gfn.actions.Actions class method) make_exit_actions() (gfn.actions.Actions class method) make_initial_states_tensor() (gfn.states.States class method) make_random_states_tensor() (gfn.states.States class method) make_sink_states_tensor() (gfn.states.States class method) make_States_class() (gfn.env.Env method) (gfn.gym.Box method) (gfn.gym.box.Box method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) maskless_backward_step() (gfn.env.Env method) (gfn.gym.Box method) (gfn.gym.box.Box method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) maskless_step() (gfn.env.Env method) (gfn.gym.Box method) (gfn.gym.box.Box method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) MaskTensor (in module gfn.gflownet.sub_trajectory_balance) max_concentration (examples.test_scripts.BoxArgs attribute) max_length (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) min_concentration (examples.test_scripts.BoxArgs attribute) ModifiedDBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.detailed_balance)

module examples examples.test_scripts examples.train_box examples.train_discreteebm examples.train_hypergrid gfn gfn.actions gfn.containers gfn.containers.base gfn.containers.replay_buffer gfn.containers.trajectories gfn.containers.transitions gfn.env gfn.gflownet gfn.gflownet.base gfn.gflownet.detailed_balance gfn.gflownet.flow_matching gfn.gflownet.sub_trajectory_balance gfn.gflownet.trajectory_balance gfn.gym gfn.gym.box gfn.gym.discrete_ebm gfn.gym.helpers gfn.gym.helpers.box_utils gfn.gym.helpers.preprocessors gfn.gym.helpers.test_box_utils gfn.gym.hypergrid gfn.modules gfn.preprocessors gfn.samplers gfn.states gfn.utils gfn.utils.common gfn.utils.distributions gfn.utils.modules module (gfn.modules.GFNModule attribute) (gfn.modules.ScalarEstimator attribute)

N

n_actions (gfn.states.DiscreteStates attribute) n_components (examples.test_scripts.BoxArgs attribute) (gfn.gym.helpers.box_utils.BoxPBNeuralNet attribute) (gfn.gym.helpers.box_utils.BoxPFNeuralNet attribute) n_components_s0 (examples.test_scripts.BoxArgs attribute) (gfn.gym.helpers.box_utils.BoxPFNeuralNet attribute) n_hidden (examples.test_scripts.CommonArgs attribute) n_states (gfn.env.DiscreteEnv property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) n_terminating_states (gfn.env.DiscreteEnv property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property)

n_trajectories (examples.test_scripts.CommonArgs attribute) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) n_transitions (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) ndim (examples.test_scripts.DiscreteEBMArgs attribute) (examples.test_scripts.HypergridArgs attribute) NeuralNet (class in gfn.utils) (class in gfn.utils.modules) next_states (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute) no_cuda (examples.test_scripts.CommonArgs attribute) NonValidActionsError (in module gfn.env) norm() (gfn.gym.Box static method) (gfn.gym.box.Box static method)

O

objects_type (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) on_policy (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute)

OneHotPreprocessor (class in gfn.gym.helpers.preprocessors) output_dim (gfn.utils.modules.DiscreteUniform attribute)

P

parser (in module examples.train_box) (in module examples.train_discreteebm) (in module examples.train_hypergrid) pb (gfn.gflownet.base.PFBasedGFlowNet attribute) (gfn.gflownet.base.TrajectoryBasedGFlowNet attribute) (gfn.gflownet.PFBasedGFlowNet attribute) (gfn.gflownet.TrajectoryBasedGFlowNet attribute) pf (gfn.gflownet.base.PFBasedGFlowNet attribute) (gfn.gflownet.base.TrajectoryBasedGFlowNet attribute) (gfn.gflownet.PFBasedGFlowNet attribute) (gfn.gflownet.TrajectoryBasedGFlowNet attribute) PFBasedGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.base)

PFs0 (gfn.gym.helpers.box_utils.BoxPFNeuralNet attribute) PI_2 (in module gfn.gym.helpers.box_utils) PI_2_INV (in module gfn.gym.helpers.box_utils) PredictionsTensor (in module gfn.gflownet.sub_trajectory_balance) preprocess() (gfn.gym.helpers.preprocessors.KHotPreprocessor method) (gfn.gym.helpers.preprocessors.OneHotPreprocessor method) (gfn.preprocessors.EnumPreprocessor method) (gfn.preprocessors.IdentityPreprocessor method) (gfn.preprocessors.Preprocessor method) Preprocessor (class in gfn.preprocessors) preprocessor (gfn.modules.GFNModule attribute), [1] (gfn.modules.ScalarEstimator attribute), [1] probability_distribution_kwargs (gfn.samplers.Sampler attribute)

Q

QuarterCircle (class in gfn.gym.helpers.box_utils)

QuarterCircleWithExit (class in gfn.gym.helpers.box_utils) QuarterDisk (class in gfn.gym.helpers.box_utils)

R

R0 (examples.test_scripts.HypergridArgs attribute) R1 (examples.test_scripts.HypergridArgs attribute) R2 (examples.test_scripts.HypergridArgs attribute) replay_buffer_size (examples.test_scripts.CommonArgs attribute) ReplayBuffer (class in gfn.containers) (class in gfn.containers.replay_buffer)

reset() (gfn.env.Env method) reward() (gfn.env.Env method) reward_matching_loss() (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method)

S

s0 (gfn.states.States attribute) sample() (gfn.containers.base.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) (gfn.gym.helpers.box_utils.DistributionWrapper method) (gfn.gym.helpers.box_utils.QuarterCircle method) (gfn.gym.helpers.box_utils.QuarterCircleWithExit method) (gfn.gym.helpers.box_utils.QuarterDisk method) (gfn.utils.distributions.UnsqueezedCategorical method) sample_actions() (gfn.samplers.Sampler method) sample_from_reward() (in module examples.train_box) sample_terminating_states() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.GFlowNet method) sample_trajectories() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.base.PFBasedGFlowNet method) (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method) (gfn.gflownet.GFlowNet method) (gfn.gflownet.PFBasedGFlowNet method) (gfn.samplers.Sampler method) Sampler (class in gfn.samplers)

save() (gfn.containers.base.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) ScalarEstimator (class in gfn.modules) scheduler_milestone (examples.test_scripts.BoxArgs attribute) seed (examples.test_scripts.CommonArgs attribute) sf (gfn.states.States attribute) sf_bias (gfn.modules.DiscretePolicyEstimator attribute) split_PF_module_output() (in module gfn.gym.helpers.box_utils) stack() (gfn.actions.Actions class method) state_shape (gfn.states.States attribute) States (class in gfn.states) states (gfn.containers.Trajectories attribute) (gfn.containers.trajectories.Trajectories attribute) (gfn.containers.Transitions attribute) (gfn.containers.transitions.Transitions attribute) step() (gfn.env.DiscreteEnv method) (gfn.env.Env method) subTB_lambda (examples.test_scripts.CommonArgs attribute) subTB_weighting (examples.test_scripts.CommonArgs attribute) SubTBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.sub_trajectory_balance)

T

table (gfn.utils.modules.Tabular attribute) (gfn.utils.Tabular attribute) Tabular (class in gfn.utils) (class in gfn.utils.modules) tabular (examples.test_scripts.CommonArgs attribute) TargetsTensor (in module gfn.gflownet.sub_trajectory_balance) TBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) temperature (gfn.modules.DiscretePolicyEstimator attribute) tensor (gfn.actions.Actions attribute) (gfn.states.States attribute) terminating_states (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) (gfn.env.DiscreteEnv property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) test_box() (in module examples.test_scripts) test_discreteebm() (in module examples.test_scripts) test_hypergrid() (in module examples.test_scripts) test_mixed_distributions() (in module gfn.gym.helpers.test_box_utils) tied (examples.test_scripts.CommonArgs attribute) to_non_initial_intermediary_and_terminating_states() (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) to_probability_distribution() (gfn.gym.helpers.box_utils.BoxPBEstimator method) (gfn.gym.helpers.box_utils.BoxPFEstimator method) (gfn.modules.DiscretePolicyEstimator method) (gfn.modules.GFNModule method)

to_states() (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) to_training_samples() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet method) (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method) (gfn.gflownet.GFlowNet method) (gfn.gflownet.ModifiedDBGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) to_transitions() (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) training_objects (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) TrainingSampleType (in module gfn.gflownet.base) Trajectories (class in gfn.containers) (class in gfn.containers.trajectories) TrajectoryBasedGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.base) Transitions (class in gfn.containers) (class in gfn.containers.transitions) true_dist_pmf (gfn.env.DiscreteEnv property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) true_reward() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method)

U

uniform_pb (examples.test_scripts.CommonArgs attribute)

UnsqueezedCategorical (class in gfn.utils.distributions) update_masks() (gfn.states.DiscreteStates method)

V

validate() (in module gfn.utils.common) validate_actions() (gfn.env.Env method)

validation_interval (examples.test_scripts.CommonArgs attribute) validation_samples (examples.test_scripts.CommonArgs attribute)

W

wandb_project (examples.test_scripts.CommonArgs attribute) weighting (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute) (gfn.gflownet.SubTBGFlowNet attribute)

when_is_done (gfn.containers.Trajectories attribute) (gfn.containers.trajectories.Trajectories attribute)