Index

_

__call__() (bench_get_scores_all._ModDBDummyEstimator method) (gfn.gflownet.HalfSquaredLoss method) (gfn.gflownet.LinexLoss method) (gfn.gflownet.losses.HalfSquaredLoss method) (gfn.gflownet.losses.LinexLoss method) (gfn.gflownet.losses.RegressionLoss method) (gfn.gflownet.losses.ShiftedCoshLoss method) (gfn.gflownet.losses.SquaredLoss method) (gfn.gflownet.RegressionLoss method) (gfn.gflownet.ShiftedCoshLoss method) (gfn.gflownet.SquaredLoss method) (gfn.gym.helpers.bayesian_structure.priors.BasePrior method) (gfn.gym.hypergrid.BitwiseXORReward method) (gfn.gym.hypergrid.ConditionalMultiScaleReward method) (gfn.gym.hypergrid.CorruptedReward method) (gfn.gym.hypergrid.CosineReward method) (gfn.gym.hypergrid.DeceptiveReward method) (gfn.gym.hypergrid.GridReward method) (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) (gfn.gym.hypergrid.OriginalReward method) (gfn.gym.hypergrid.SparseReward method) (gfn.gym.hypergrid.UniformRandomReward method) (gfn.preprocessors.Preprocessor method) (tutorials.examples.train_graph_ring.RingReward method) (tutorials.examples.train_graph_triangle.TriangleReward method) __del__() (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) __enter__() (gfn.utils.common.Timer method) __eq__() (gfn.gflownet.LinexLoss method) (gfn.gflownet.losses.LinexLoss method) (gfn.gflownet.losses.RegressionLoss method) (gfn.gflownet.RegressionLoss method) __exit__() (gfn.utils.common.Timer method) __getattr__() (gfn.gym.helpers.chip_design.plc_client.PlacementCost method) __getitem__() (bench_get_scores_all._DBDummyActions method) (bench_get_scores_all._DBDummyStates method) (bench_get_scores_all._ModDBDummyActions method) (bench_get_scores_all._ModDBDummyStates method) (bench_get_scores_all._ModDBDummyTransitions method) (gfn.actions.Actions method) (gfn.containers.base.Container method) (gfn.containers.Container method) (gfn.containers.replay_buffer.Container method) (gfn.containers.states_container.StatesContainer method) (gfn.containers.StatesContainer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.chip_design.ChipDesignStates method) (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) __hash__() (gfn.gflownet.LinexLoss method) (gfn.gflownet.losses.LinexLoss method) (gfn.gflownet.losses.RegressionLoss method) (gfn.gflownet.RegressionLoss method) __len__() (bench_get_scores_all._DBDummyActions method) (bench_get_scores_all._DBDummyStates method) (bench_get_scores_all._DBDummyTransitions method) (bench_get_scores_all._ModDBDummyActions method) (bench_get_scores_all._ModDBDummyStates method) (bench_get_scores_all._ModDBDummyTransitions method) (bench_get_scores_all._SubTBDummyTrajectories method) (gfn.actions.Actions method) (gfn.containers.base.Container method) (gfn.containers.Container method) (gfn.containers.replay_buffer.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) (gfn.containers.states_container.StatesContainer method) (gfn.containers.StatesContainer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.states.GraphStates method) (gfn.states.States method) __repr__() (gfn.actions.Actions method) (gfn.actions.GraphActions method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) (gfn.containers.states_container.StatesContainer method) (gfn.containers.StatesContainer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.estimators.Estimator method) (gfn.gflownet.LinexLoss method) (gfn.gflownet.losses.LinexLoss method) (gfn.gflownet.losses.RegressionLoss method) (gfn.gflownet.RegressionLoss method) (gfn.preprocessors.Preprocessor method) (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) __setitem__() (gfn.actions.Actions method) (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.chip_design.ChipDesignStates method) (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) __slots__ (gfn.estimators.RolloutContext attribute) __version__ (in module gfn) _add_counter (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _add_logz_diff() (gfn.env.DiscreteEnv method) _all_states_tensor (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) _apply_large_model_scaling() (in module tutorials.examples.train_with_compile) _apply_parity_checks() (gfn.gym.hypergrid.BitwiseXORReward method) _apply_state_to_plc() (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) _assert_factory_accepts_debug() (in module gfn.states) _assert_no_nan() (gfn.gflownet.mle.MLEDiffusion method) _AutoregressiveTransformerBlock (class in gfn.utils.modules) _B (gfn.gym.hypergrid.BitwiseXORReward attribute) _backward_masks_cache (gfn.states.DiscreteStates attribute) _backward_step() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) _baseline_ema (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) _baseline_kept (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _baseline_log_reward (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _baseline_skipped_sends (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _baseline_total (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _beta_mixture_log_prob() (gfn.gym.helpers.box_cartesian_utils._BetaMixtureMixin method) _BetaMixtureMixin (class in gfn.gym.helpers.box_cartesian_utils) _bit_positions (gfn.gym.hypergrid.BitwiseXORReward attribute) _block_log_density() (gfn.gym.diffusion_sampling.ManyWell static method) _BoxCartesianMLP (class in gfn.gym.helpers.box_cartesian_utils) _buffer (gfn.utils.distributed.AsyncSendHandle attribute) _build_bitsequence_mlp_components() (in module tutorials.examples.train_with_compile) _build_bitsequence_recurrent_components() (in module tutorials.examples.train_with_compile) _build_diffusion_components() (in module tutorials.examples.train_with_compile) _build_graph_ring_components() (in module tutorials.examples.train_with_compile) _build_hypergrid_components() (in module tutorials.examples.train_with_compile) _build_line_components() (in module tutorials.examples.train_with_compile) _build_lpv() (in module bench_get_scores_all) _build_tb() (in module bench_get_scores_all) _build_tree() (gfn.gym.perfect_tree.PerfectBinaryTree method) (gfn.gym.PerfectBinaryTree method) _calculate_bge_score() (gfn.gym.helpers.bayesian_structure.scores.BGeScore method) _calculate_module_output() (gfn.estimators.ConditionalLogZEstimator method) (gfn.estimators.ScalarEstimator method) _call_estimator_with_conditions() (in module gfn.gflownet.detailed_balance) _call_simple() (gfn.gym.hypergrid.CorruptedReward method) _collect_pending_score() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _combine_prev_and_recon_trajectories() (gfn.samplers.LocalSearchSampler static method) _comm_stats (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) _compare() (gfn.actions.Actions method) (gfn.states.GraphStates method) (gfn.states.States method) _compare_reference() (gfn.states.GraphStates method) _compute_backward_masks() (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequenceStates method) (gfn.states.DiscreteStates method) _compute_envelope_k() (gfn.gym.diffusion_sampling.ManyWell method) _compute_forward_masks() (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequenceStates method) (gfn.states.DiscreteStates method) _compute_logits_for_distribution() (gfn.estimators.LogitBasedEstimator static method) _compute_metadata() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _compute_rtb_scores() (gfn.gflownet.RelativeTBBase method) (gfn.gflownet.trajectory_balance.RelativeTBBase method) _conditions (gfn.states.GraphStates attribute) (gfn.states.States attribute) _consecutive_filtered_empty (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _db_build_model_and_data() (in module bench_get_scores_all) _db_original_get_scores() (in module bench_get_scores_all) _DBDummyActions (class in bench_get_scores_all) _DBDummyEnv (class in bench_get_scores_all) _DBDummyStates (class in bench_get_scores_all) _DBDummyTransitions (class in bench_get_scores_all) _decode_action() (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.NonAutoregressiveBitSequence method) _DEF_FILE (in module gfn.gym.helpers.chip_design.build_container) _device (env.CodonDesignEnv attribute) (gfn.states.GraphStates attribute), [1] _dim_idx (gfn.gym.hypergrid.BitwiseXORReward attribute) _DIR (in module gfn.gym.helpers.chip_design.build_container) _diversity_repr() (gfn.containers.NormBasedDiversePrioritizedReplayBuffer static method) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer static method) _DOCKER_URI (in module gfn.gym.helpers.chip_design.build_container) _edges_dim (gfn.utils.modules.GraphEdgeActionMLP attribute) _enumerate_all_states_tensor() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _EPS (gfn.gym.hypergrid.GridReward attribute) _estimate_cost_stats() (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) _estimate_replacement_rates() (gfn.gym.hypergrid.CorruptedReward method) _even_parity_mask() (gfn.gym.hypergrid.BitwiseXORReward method) _exists_bitwise_xor() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_conditional_multiscale() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_cosine() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_fallback_random() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_multiplicative_coprime() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_original_or_deceptive() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_random_or_corrupted() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _exists_sparse() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _extract_digits() (gfn.gym.hypergrid.ConditionalMultiScaleReward method) _extract_samples() (gfn.gflownet.mle.MLEDiffusion method) _factor_exponents_up_to_cap() (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) _filter_and_send() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _filter_by_baseline() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _find_singularity() (in module gfn.gym.helpers.chip_design.build_container) (in module gfn.gym.helpers.chip_design.plc_client)

_first_env() (in module gfn.utils.distributed) _first_k_dims() (in module gfn.gym.hypergrid) _flush_pending() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _format_ms() (in module bench_get_scores_all) _forward_masks_cache (gfn.states.DiscreteStates attribute) _forward_trunk() (gfn.estimators.ConditionalDiscretePolicyEstimator method) _generate_combinations_in_batches() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _generate_rule_targets() (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) _get_index_np() (gfn.states.GraphStates method) _get_MPI() (in module gfn.utils.distributed) _get_states_indices_bigint() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _gf2_random_fullrank() (in module gfn.gym.hypergrid) _gf2_rank() (in module gfn.gym.hypergrid) _group_mean() (gfn.utils.modules.GraphActionGNN static method) _handle_message_async() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _handle_message_sync() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _hard_macro_indices (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) _head_A_per_rule (gfn.gym.hypergrid.BitwiseXORReward attribute) _head_c_per_rule (gfn.gym.hypergrid.BitwiseXORReward attribute) _helix_lookups() (MFE_calculator.RNAFolderBase static method) _hypergrid_worker() (in module gfn.gym.hypergrid) _inject_baseline_log_reward() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _input_dim (gfn.gym.helpers.box_polar_utils.BoxPBMLP attribute) (gfn.gym.helpers.box_polar_utils.BoxPFMLP attribute) (gfn.gym.helpers.box_utils.BoxPBMLP attribute) (gfn.gym.helpers.box_utils.BoxPFMLP attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) (gfn.utils.modules.MLP attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) _INT_PAIRS (in module MFE_utils) _integers_to_binary() (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence static method) (gfn.gym.NonAutoregressiveBitSequence static method) _interpolated_table() (MFE_calculator.RNAFolderBase static method) _invalidate_masks_cache() (gfn.states.DiscreteStates method) _is_full (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _is_initial_cache (gfn.states.States attribute) _is_sink_cache (gfn.states.States attribute) _is_tiered (gfn.gym.hypergrid.CorruptedReward attribute) _isend_and_defer_score() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _jsd() (gfn.env.DiscreteEnv static method) _lcm_ok() (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) _load_reference_sequences_from_file() (CAI_calculator.CAICalculator method) _local_add() (gfn.containers.NormBasedDiversePrioritizedReplayBuffer method) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.replay_buffer.TerminatingStateBuffer method) (gfn.containers.ReplayBuffer method) (gfn.containers.TerminatingStateBuffer method) _log_action (bench_get_scores_all._ModDBDummyEstimator attribute) (bench_get_scores_all._ModDBFakeDist attribute) _log_exit (bench_get_scores_all._ModDBDummyEstimator attribute) (bench_get_scores_all._ModDBFakeDist attribute) _log_partition (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) _log_partition_cache (gfn.gym.ConditionalHyperGrid attribute) (gfn.gym.hypergrid.ConditionalHyperGrid attribute) _log_prior (gfn.gym.helpers.bayesian_structure.priors.BasePrior attribute) _log_reward_fn (bench_get_scores_all._DBDummyEnv attribute) _log_rewards (bench_get_scores_all._TBTrajectoriesStub attribute) (gfn.containers.states_container.StatesContainer attribute), [1] (gfn.containers.StatesContainer attribute), [1] (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] _loop_lookups() (MFE_calculator.RNAFolderBase static method) _lpv_original_loss() (in module bench_get_scores_all) _make_modes() (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.NonAutoregressiveBitSequence method) _make_optimizer_for() (in module tutorials.examples.train_hypergrid) _make_proposal() (gfn.gym.diffusion_sampling.ManyWell method) _make_view() (gfn.states.DiscreteStates class method) (gfn.states.States class method) _MAX_POOL_WORKERS (in module gfn.gym.hypergrid) _max_reward (gfn.gym.ConditionalHyperGrid attribute) (gfn.gym.hypergrid.ConditionalHyperGrid attribute) _maybe_compile() (in module bench_get_scores_all) _merge_conditions() (gfn.states.States method) _min_hamming_distance() (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence static method) (gfn.gym.NonAutoregressiveBitSequence static method) _mix_with_uniform_in_log_space() (gfn.estimators.LogitBasedEstimator static method) _moddb_build_model_and_data() (in module bench_get_scores_all) _moddb_original_get_scores() (in module bench_get_scores_all) _ModDBDummyActions (class in bench_get_scores_all) _ModDBDummyEstimator (class in bench_get_scores_all) _ModDBDummyStates (class in bench_get_scores_all) _ModDBDummyTransitions (class in bench_get_scores_all) _ModDBFakeDist (class in bench_get_scores_all) _mode_reward_threshold() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _mode_stats_kind (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) _modes_exist_quick_check() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _modes_exist_quick_check_info() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) _mps_backend_available() (in module tutorials.examples.train_with_compile) _n_comp_max (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPFMLP attribute) (gfn.gym.helpers.box_utils.BoxPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxPFMLP attribute) _n_mode_states_estimate (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) _n_mode_states_exact (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) _n_nodes (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) _normalize_cost() (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) _normalize_keys() (in module tutorials.examples.train_with_compile) _num_variables (gfn.gym.helpers.bayesian_structure.priors.BasePrior attribute) _original_reward_fn (gfn.gym.ConditionalHyperGrid attribute) (gfn.gym.hypergrid.ConditionalHyperGrid attribute) _output_dim (gfn.utils.modules.GraphEdgeActionMLP attribute) (gfn.utils.modules.MLP attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) _output_shape (gfn.gym.helpers.box_polar_utils.DistributionWrapper attribute) (gfn.gym.helpers.box_utils.DistributionWrapper attribute) _PAIRS (in module MFE_calculator) (in module MFE_utils) _PAIRS_INV (in module MFE_utils) _pairwise_coprime_ok() (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) _parse_args() (in module bench_get_scores_all) _pe (gfn.utils.modules.SinusoidalPositionalEmbedding attribute) _pending_batches (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _pending_len (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _pending_score (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _pending_sends (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) _per_rule_mode_counts() (gfn.gym.hypergrid.BitwiseXORReward method) _PKG_DIR (in module gfn.gym.helpers.chip_design.plc_client) _POLICY_REQUIRED_METHODS (in module gfn.estimators) _positional_embedding_type (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) _prepare_for_remote() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _prepare_logits() (gfn.estimators.LogitBasedEstimator static method) _preset_seed() (in module gfn.gym.hypergrid) _print_mode_stats() (in module tutorials.examples.train_hypergrid_exploration_examples) _print_timing_summary() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _proxy_cost() (in module gfn.gym.helpers.chip_design.utils) _prune_completed_sends() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _recv_object() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) _recv_score() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _rejection_sampling_x1() (gfn.gym.diffusion_sampling.ManyWell static method) _replacement_rates (gfn.gym.hypergrid.CorruptedReward attribute) _request (gfn.utils.distributed.AsyncSendHandle attribute) _resolve_plc_binary() (in module gfn.gym.helpers.chip_design.plc_client) _REWARD_CLASSES (gfn.gym.hypergrid.CorruptedReward attribute) _RULE_SEED_STRIDE (gfn.gym.hypergrid.BitwiseXORReward attribute) (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) _rule_target_exps (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) _run_db() (in module bench_get_scores_all) _run_moddb() (in module bench_get_scores_all) _run_subtb() (in module bench_get_scores_all) _run_tb_or_lpv() (in module bench_get_scores_all) _run_with_compile_variants() (in module bench_get_scores_all) _sample_beta_mixture() (gfn.gym.helpers.box_cartesian_utils._BetaMixtureMixin method) _scale_int() (in module bench_get_scores_all) _scale_noise() (gfn.utils.modules.NoisyLinear method) _scale_pair() (in module bench_get_scores_all) _select_dtype() (in module bench_get_scores_all) _select_powers (gfn.gym.hypergrid.BitwiseXORReward attribute) _selector() (gfn.gym.hypergrid.ConditionalMultiScaleReward method) (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) _send_data() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _send_handle (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) _send_objs() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _send_objs_async() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _shift_coeffs_tensor (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) _SIF_FILE (in module gfn.gym.helpers.chip_design.build_container) _solve_gf2_has_solution() (gfn.gym.HyperGrid static method) (gfn.gym.hypergrid.HyperGrid static method) _solve_gf2_witness() (gfn.gym.HyperGrid static method) (gfn.gym.hypergrid.HyperGrid static method) _sort_and_truncate() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _sorted_node_indices (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) _special_lookups() (MFE_calculator.RNAFolderBase static method) _state_hash_uniform() (in module gfn.gym.hypergrid) _step() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) _subtb_build_model_and_data() (in module bench_get_scores_all) _subtb_original_get_scores() (in module bench_get_scores_all) _SubTBDummyTrajectories (class in bench_get_scores_all) _summarize_iteration_times() (in module tutorials.examples.train_with_compile) _tb_original_get_scores() (in module bench_get_scores_all) _tb_original_loss() (in module bench_get_scores_all) _TBTrajectoriesStub (class in bench_get_scores_all) _tier_check_counts (gfn.gym.hypergrid.BitwiseXORReward attribute) _tier_weights_t (gfn.gym.hypergrid.BitwiseXORReward attribute) _time_fn() (in module bench_get_scores_all) _timing_data (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) _torch_dynamo (in module tutorials.examples.train_with_compile) _true_dist (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) _true_dist_cache (gfn.gym.ConditionalHyperGrid attribute) (gfn.gym.hypergrid.ConditionalHyperGrid attribute) _uniform_log_probs() (gfn.estimators.LogitBasedEstimator static method) _uniform_partition (gfn.gym.hypergrid.BitwiseXORReward attribute) (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) _update_baseline() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _validate_rule_coverage() (gfn.gym.hypergrid.BitwiseXORReward method) (gfn.gym.hypergrid.ConditionalMultiScaleReward method) (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) _vocab_size (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) _wait_previous_send() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) _warn_if_insufficient_samples() (gfn.env.DiscreteEnv method)

A

ACTION_INDICES (gfn.actions.GraphActions attribute), [1] action_shape (gfn.actions.Actions attribute), [1] (gfn.actions.GraphActions attribute) (gfn.env.Env attribute), [1] action_type (gfn.actions.GraphActions property) ACTION_TYPE_KEY (gfn.actions.GraphActions attribute), [1] action_type_mlp (gfn.utils.modules.GraphActionGNN attribute) actions (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyTransitions attribute) Actions (class in gfn.actions) actions (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] Actions (gfn.env.DiscreteEnv attribute) (gfn.env.Env attribute), [1] (gfn.env.GraphEnv attribute), [1] actions_detach (gfn.utils.distributions.IsotropicGaussian attribute), [1] actions_from_batch_shape() (gfn.env.Env method) actions_from_tensor() (gfn.env.Env method) ACTIVATION_FNS (in module gfn.utils.modules) active_dims (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) add() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) add_arg_group() (in module tutorials.examples.train_diffusion_rtb) ADD_EDGE (gfn.actions.GraphActionType attribute) ADD_NODE (gfn.actions.GraphActionType attribute) agent_group_id (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) agent_group_size (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) agent_groups (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) ALL_CODONS (in module utils) all_dags() (in module gfn.gym.helpers.bayesian_structure.jsd) all_gather() (in module gfn.utils.distributed) all_indices() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) all_log_rewards (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) all_reduce() (in module gfn.utils.distributed) 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) (gfn.gym.perfect_tree.PerfectBinaryTree property) (gfn.gym.PerfectBinaryTree property) (gfn.gym.set_addition.SetAddition property) (gfn.gym.SetAddition property)

alpha (gfn.gflownet.flow_matching.FMGFlowNet attribute), [1] (gfn.gflownet.FMGFlowNet attribute), [1] (gfn.gflownet.LinexLoss attribute) (gfn.gflownet.losses.LinexLoss attribute) (gfn.gflownet.trajectory_balance.TrustPCLGFlowNet property) (gfn.gflownet.TrustPCLGFlowNet property) (gfn.gym.discrete_ebm.DiscreteEBM attribute), [1] (gfn.gym.DiscreteEBM attribute), [1] (gfn.gym.helpers.box_cartesian_utils._BetaMixtureMixin attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) alpha_mu (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) alpha_w (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) analytic_log_partition() (gfn.gym.hypergrid.ConditionalMultiScaleReward method) analytic_mode_count() (gfn.gym.hypergrid.BitwiseXORReward method) (gfn.gym.hypergrid.ConditionalMultiScaleReward method) analytical_placer() (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) any_dim_near_origin (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) append() (gfn.gym.helpers.bayesian_structure.jsd.GraphCollection method) arg_constraints (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) as_dots() (MFE_calculator.RNASolution method) assert_finite_gradients() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.GFlowNet method) assert_finite_parameters() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.GFlowNet method) assigned_buffer (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) assigned_training_ranks (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) async_comm (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) async_score (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) AsyncSendHandle (class in gfn.utils.distributed) at_boundary (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) attention (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) attn_dropout (gfn.utils.modules._AutoregressiveTransformerBlock attribute) AutoregressiveDiscreteSequenceModel (class in gfn.utils.modules) average_gradients() (in module gfn.utils.distributed) average_models() (in module gfn.utils.distributed)

B

backbone (gfn.utils.modules.GraphScalarMLP attribute) backward (tutorials.examples.train_line.ScaledGaussianWithOptionalExit attribute) (tutorials.examples.train_line.StepEstimator attribute) backward_masks (gfn.states.DiscreteStates attribute) (gfn.states.DiscreteStates property) (gfn.states.GraphStates property) backward_sampler (gfn.samplers.LocalSearchSampler attribute), [1] backward_step() (env.CodonDesignEnv method) (gfn.env.Env method) (gfn.env.GraphEnv method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequence.BitSequencePlus method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.BitSequencePlus method) (gfn.gym.box.BoxPolar method) (gfn.gym.BoxPolar method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) (gfn.gym.diffusion_sampling.DiffusionSampling method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.GraphBuilding method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.Line method) (gfn.gym.line.Line method) (gfn.gym.NonAutoregressiveBitSequence method) (gfn.gym.perfect_tree.PerfectBinaryTree method) (gfn.gym.PerfectBinaryTree method) (gfn.gym.set_addition.SetAddition method) (gfn.gym.SetAddition method) barrier() (in module gfn.utils.distributed) base (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.utils.modules.SinusoidalPositionalEmbedding attribute) base_as_int() (in module MFE_utils) base_dist (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] base_fn (gfn.gym.hypergrid.CorruptedReward attribute) base_r_dist (gfn.gym.helpers.box_polar_utils.QuarterDisk attribute), [1] (gfn.gym.helpers.box_utils.QuarterDisk attribute), [1] base_reward_str (gfn.gym.hypergrid.CorruptedReward attribute) base_theta_dist (gfn.gym.helpers.box_polar_utils.QuarterDisk attribute), [1] (gfn.gym.helpers.box_utils.QuarterDisk attribute), [1] baseline_ema_alpha (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) baseline_filtering (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) baseline_percentile (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) baseline_refresh_after (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) baseline_strategy (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) BasePrior (class in gfn.gym.helpers.bayesian_structure.priors) BaseScore (class in gfn.gym.helpers.bayesian_structure.scores) BaseTarget (class in gfn.gym.diffusion_sampling) batch_ptrs (gfn.utils.graphs.GeometricBatch attribute) batch_shape (bench_get_scores_all._DBDummyActions property) (bench_get_scores_all._DBDummyStates property) (gfn.actions.Actions property) (gfn.actions.GraphActions property) (gfn.states.GraphStates property) (gfn.states.States property) (gfn.utils.graphs.GeometricBatch attribute) batch_size (bench_get_scores_all._SubTBDummyTrajectories attribute) (bench_get_scores_all._TBTrajectoriesStub attribute) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) (gfn.estimators.RolloutContext attribute) (performance_tuning.BenchmarkSettings attribute) BayesianStructure (class in gfn.gym.bayesian_structure)

bench_get_scores_all module BenchmarkConfig (class in performance_tuning) BenchmarkResult (class in performance_tuning) BenchmarkSettings (class in performance_tuning) beta (gfn.gym.helpers.bayesian_structure.priors.EdgePrior attribute) (gfn.gym.helpers.box_cartesian_utils._BetaMixtureMixin attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) BGeScore (class in gfn.gym.helpers.bayesian_structure.scores) bias (gfn.utils.modules.NoisyLinear property) binary_to_integers() (gfn.gym.BitSequence static method) (gfn.gym.bitSequence.BitSequence static method) bits_per_tier (gfn.gym.hypergrid.BitwiseXORReward attribute) BitSequence (class in gfn.gym) (class in gfn.gym.bitSequence) BitSequencePlus (class in gfn.gym) (class in gfn.gym.bitSequence) BitSequenceStates (class in gfn.gym.bitSequence) BitwiseXORReward (class in gfn.gym.hypergrid) Box (in module gfn.gym) BoxCartesian (class in gfn.gym) (class in gfn.gym.box_cartesian) BoxCartesianDistribution (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) BoxCartesianPBDistribution (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) BoxCartesianPBEstimator (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) BoxCartesianPBMLP (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) BoxCartesianPFEstimator (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) BoxCartesianPFMLP (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) BoxPBEstimator (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) BoxPBMLP (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) BoxPBUniform (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) BoxPFEstimator (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) BoxPFMLP (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) BoxPolar (class in gfn.gym) (class in gfn.gym.box) BoxStateFlowModule (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) branches() (MFE_calculator.RNASolutionBase2 method) branching_factor (gfn.gym.perfect_tree.PerfectBinaryTree attribute), [1] (gfn.gym.PerfectBinaryTree attribute), [1] broadcast() (in module gfn.utils.distributed) bts_logits_scaled (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) buffer_group (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) BUFFER_LEN (gfn.gym.helpers.chip_design.plc_client.PlacementCost attribute) build() (in module gfn.gym.helpers.chip_design.build_container) build_components() (in module performance_tuning) build_conditional_logF_scalar_estimator() (in module tutorials.examples.train_conditional) build_conditional_pf_pb() (in module tutorials.examples.train_conditional) build_db_gflownet() (in module tutorials.examples.train_conditional) build_db_mod_gflownet() (in module tutorials.examples.train_conditional) build_fm_gflownet() (in module tutorials.examples.train_conditional) build_forward_estimator() (in module tutorials.examples.train_diffusion_rtb) build_gflownet() (in module tutorials.examples.train_hypergrid_exploration_examples) build_subTB_gflownet() (in module tutorials.examples.train_conditional) build_tb_gflownet() (in module tutorials.examples.train_conditional) builder (tutorials.examples.train_with_compile.EnvironmentBenchmark attribute)

C

cached_sample() (gfn.gym.diffusion_sampling.BaseTarget method) CAI_calculator module CAICalculator (class in CAI_calculator) 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_partition (gfn.gym.HyperGrid attribute), [1] (gfn.gym.hypergrid.HyperGrid attribute), [1] 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.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.ReplayBuffer attribute), [1] (gfn.containers.replay_buffer.TerminatingStateBuffer attribute) (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) (gfn.containers.ReplayBuffer attribute), [1] (gfn.containers.TerminatingStateBuffer attribute) carry (gfn.estimators.RolloutContext attribute) categorical_edge_features (gfn.states.GraphStates attribute) categorical_node_features (gfn.states.GraphStates attribute) cd_finetune (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) centers (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1] check_compatibility() (in module gfn.gflownet.detailed_balance) ChipDesign (class in gfn.gym) (class in gfn.gym.chip_design) ChipDesignPreprocessor (class in tutorials.examples.train_chip_design) (class in tutorials.examples.train_chip_design_medium) ChipDesignStates (class in gfn.gym.chip_design) CLAMP (in module gfn.gym.helpers.box_polar_utils) cleanup() (gfn.utils.distributed.DistributedContext method) clipping (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute) clone() (gfn.actions.Actions method) (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.chip_design.ChipDesignStates method) (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) close() (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) (gfn.gym.helpers.chip_design.plc_client.PlacementCost method) closed_energy() (MFE_calculator.RNAFolderBase2 method) closures (gfn.gym.helpers.bayesian_structure.jsd.FullPosterior attribute) codon_gc_counts (env.CodonDesignEnv attribute) (in module utils) CODON_TABLE (in module utils) CODON_TO_IDX (in module utils) CodonDesignEnv (class in env) CodonSequencePreprocessor (class in preprocessor) color (tutorials.examples.train_with_compile.EnvironmentBenchmark attribute) column_names (gfn.gym.helpers.bayesian_structure.scores.BaseScore attribute) communication_backend (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) (gfn.containers.ReplayBuffer attribute) compare_data_objects() (in module gfn.utils.graphs) compile_estimators (tutorials.examples.train_with_compile.CompileMode attribute) compile_loss (tutorials.examples.train_with_compile.CompileMode attribute) COMPILE_MODE_COLORS (in module tutorials.examples.train_with_compile) COMPILE_MODES (in module tutorials.examples.train_with_compile) compiled (performance_tuning.BenchmarkResult attribute) CompileMode (class in tutorials.examples.train_with_compile) component_mix (gfn.gym.diffusion_sampling.ManyWell attribute) compute_cai() (CAI_calculator.CAICalculator method) (in module utils)

compute_dist() (gfn.estimators.PolicyEstimatorProtocol method) (gfn.estimators.PolicyMixin method) (gfn.estimators.RecurrentPolicyMixin method) compute_gc_content_vectorized() (in module utils) compute_intrinsic_reward() (tutorials.examples.train_hypergrid_gafn.RND method) compute_mfe_energy() (in module utils) compute_reward() (in module utils) compute_reward_components() (in module utils) compute_rnd_loss() (tutorials.examples.train_hypergrid_gafn.RND method) condition_dim (gfn.gym.ConditionalHyperGrid attribute) (gfn.gym.hypergrid.ConditionalHyperGrid attribute) condition_module (gfn.estimators.ConditionalDiscretePolicyEstimator attribute), [1] (gfn.estimators.ConditionalScalarEstimator attribute) ConditionalDiscretePolicyEstimator (class in gfn.estimators) ConditionalHyperGrid (class in gfn.gym) (class in gfn.gym.hypergrid) ConditionalLogZEstimator (class in gfn.estimators) ConditionalMultiScaleReward (class in gfn.gym.hypergrid) ConditionalScalarEstimator (class in gfn.estimators) conditions (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyTransitions attribute) (bench_get_scores_all._TBTrajectoriesStub attribute) (gfn.estimators.RolloutContext attribute) (gfn.states.States property) config (in module CAI_calculator) (performance_tuning.BenchmarkResult attribute) congestion_weight (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) constant_pb (gfn.gflownet.base.PFBasedGFlowNet attribute), [1] (gfn.gflownet.base.TrajectoryBasedGFlowNet attribute) (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet attribute) (gfn.gflownet.LogPartitionVarianceGFlowNet attribute) (gfn.gflownet.ModifiedDBGFlowNet attribute) (gfn.gflownet.PFBasedGFlowNet attribute), [1] (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) (gfn.gflownet.TrajectoryBasedGFlowNet attribute) construct_state_dag_with_bfs() (in module gfn.gym.helpers.bayesian_structure.jsd) Container (class in gfn.containers) (class in gfn.containers.base) (class in gfn.containers.replay_buffer) ContainerUnion (in module gfn.containers.replay_buffer) ContributionsTensor (in module gfn.gflownet.sub_trajectory_balance) conv_blks (gfn.utils.modules.GraphActionGNN attribute) coordinator_rank (gfn.utils.distributed.DistributedContext attribute) coprime_pairs (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) coprime_start_tier (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) CorruptedReward (class in gfn.gym.hypergrid) corruption_rate (gfn.gym.hypergrid.CorruptedReward attribute) CosineReward (class in gfn.gym.hypergrid) COST_COMPONENTS (in module gfn.gym.helpers.chip_design.utils) cost_info_function() (in module gfn.gym.helpers.chip_design.utils) CostStats (class in gfn.gym.chip_design) count_recovered_modes() (in module tutorials.examples.train_with_example_modes) create_blockages_by_spacing_constraints() (in module gfn.gym.helpers.chip_design.utils) create_placement_cost() (in module gfn.gym.helpers.chip_design.utils) create_test_set() (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) cross_dim_mods (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) cumulative_logprobs() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) CumulativeLogProbsTensor (in module gfn.gflownet.sub_trajectory_balance) current_estimator_output (gfn.estimators.RolloutContext attribute) current_node_idx (gfn.gym.chip_design.ChipDesignStates attribute) cutoff_distance (gfn.containers.NormBasedDiversePrioritizedReplayBuffer attribute), [1] (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute), [1]

D

DAGEdgeActionGNN (class in tutorials.examples.train_bayesian_structure) DAGEdgeActionGNNv2 (class in tutorials.examples.train_bayesian_structure) DAGEdgeActionMLP (class in tutorials.examples.train_bayesian_structure) DATA (gfn.containers.message.MessageType attribute) data (gfn.gym.helpers.bayesian_structure.scores.BaseScore attribute) (gfn.states.GraphStates attribute), [1] data_share_storage() (in module gfn.utils.graphs) DATA_TAG (in module gfn.containers.replay_buffer_manager) DBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.detailed_balance) dc_mpi4py (gfn.utils.distributed.DistributedContext attribute) debug (gfn.actions.Actions attribute) (gfn.actions.GraphActions attribute) (gfn.env.Env attribute) (gfn.env.GraphEnv attribute) (gfn.estimators.Estimator attribute) (gfn.gflownet.base.GFlowNet attribute) (gfn.gflownet.GFlowNet attribute) (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.gym.helpers.box_polar_utils.DistributionWrapper attribute) (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute) (gfn.gym.helpers.box_polar_utils.QuarterDisk attribute) (gfn.gym.helpers.box_utils.DistributionWrapper attribute) (gfn.gym.helpers.box_utils.QuarterCircle attribute) (gfn.gym.helpers.box_utils.QuarterDisk attribute) (gfn.states.GraphStates attribute) (gfn.states.States attribute) (gfn.utils.distributions.GraphActionDistribution attribute) (gfn.utils.distributions.UnsqueezedCategorical attribute) (performance_tuning.BenchmarkConfig attribute) DeceptiveReward (class in gfn.gym.hypergrid) decode_sequence() (in module utils) def_refs (in module CAI_calculator) default_backend (in module gfn.utils.distributed) DEFAULT_COMPILE_ORDER (in module tutorials.examples.train_with_compile) default_fill_value_for_dtype() (in module gfn.utils.common) DEFAULT_FLOW_ORDER (in module tutorials.examples.train_with_compile) default_scoring_function() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) DEFAULT_SEED (in module tutorials.examples.train_bit_sequences) (in module tutorials.examples.train_bitsequence_non_autoregressive) (in module tutorials.examples.train_box) (in module tutorials.examples.train_box_legacy) (in module tutorials.examples.train_conditional) (in module tutorials.examples.train_discreteebm) DEFAULT_SIF_IMAGE (in module gfn.gym.helpers.chip_design) (in module gfn.gym.helpers.chip_design.plc_client) delta (gfn.gym.box.BoxPolar attribute), [1] (gfn.gym.BoxPolar attribute), [1] (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute), [1] (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterDisk attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.QuarterDisk attribute), [1] density_metrics() (gfn.gym.diffusion_sampling.DiffusionSampling static method) density_weight (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) depth (gfn.gym.perfect_tree.PerfectBinaryTree attribute), [1] (gfn.gym.PerfectBinaryTree attribute), [1] description (tutorials.examples.train_with_compile.CompileMode attribute) (tutorials.examples.train_with_compile.EnvironmentBenchmark attribute) (tutorials.examples.train_with_compile.FlowVariant attribute) deserialize() (gfn.containers.message.Message static method) deterministic_mode (performance_tuning.BenchmarkSettings attribute) device (bench_get_scores_all._DBDummyStates property) (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyStates property) (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.actions.Actions property) (gfn.containers.base.Container property) (gfn.containers.Container property) (gfn.containers.replay_buffer.ReplayBuffer property) (gfn.containers.ReplayBuffer property) (gfn.containers.states_container.StatesContainer property) (gfn.containers.StatesContainer property) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) (gfn.env.Env property) (gfn.env.GraphEnv property) (gfn.estimators.RolloutContext attribute) (gfn.gym.box.BoxPolar attribute) (gfn.gym.BoxPolar attribute) (gfn.gym.diffusion_sampling.BaseTarget attribute), [1] (gfn.gym.diffusion_sampling.DiffusionSampling attribute) (gfn.states.DiscreteStates attribute) (gfn.states.GraphStates property) (gfn.states.States property) (gfn.utils.modules.Tabular attribute), [1] (performance_tuning.BenchmarkSettings attribute) (preprocessor.CodonSequencePreprocessor attribute) (tutorials.examples.train_graph_ring.RingReward attribute) (tutorials.examples.train_graph_triangle.TriangleReward attribute)

device_str (gfn.gym.BitSequence attribute) (gfn.gym.bitSequence.BitSequence attribute) DIFFUSION_TARGETS (gfn.gym.diffusion_sampling.DiffusionSampling attribute) DIFFUSION_TERMINAL_TIME_EPS (in module gfn.constants) DiffusionFixedBackwardModule (class in gfn.utils.modules) DiffusionPISGradNetBackward (class in gfn.utils.modules) DiffusionPISGradNetForward (class in gfn.utils.modules) DiffusionPISJointPolicy (class in gfn.utils.modules) DiffusionPISStateEncoding (class in gfn.utils.modules) DiffusionPISTimeEncoding (class in gfn.utils.modules) DiffusionPolicyEstimator (class in gfn.estimators) DiffusionSampling (class in gfn.gym.diffusion_sampling) dim (gfn.gym.diffusion_sampling.BaseTarget attribute), [1] (gfn.gym.diffusion_sampling.DiffusionSampling attribute) dim_near_origin (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) dims_constrained (gfn.gym.hypergrid.BitwiseXORReward attribute) directed (tutorials.examples.train_graph_ring.RingReward attribute) directed_reward() (tutorials.examples.train_graph_ring.RingReward method) disconnect_high_fanout_nets() (in module gfn.gym.helpers.chip_design.utils) DiscreteEBM (class in gfn.gym) (class in gfn.gym.discrete_ebm) DiscreteEnv (class in gfn.env) DiscreteGraphPolicyEstimator (class in gfn.estimators) DiscretePolicyEstimator (class in gfn.estimators) DiscreteStates (class in gfn.states) DiscreteUniform (class in gfn.utils.modules) dist (tutorials.examples.train_line.ScaledGaussianWithOptionalExit attribute) dist_dominant (gfn.gym.diffusion_sampling.Funnel attribute) dist_without_exit (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1] DistributedContext (class in gfn.utils.distributed) DistributedContextMPI4Py (class in gfn.utils.distributed) distribution (gfn.gym.diffusion_sampling.SimpleGaussianMixture attribute) DistributionWrapper (class in gfn.gym.helpers.box_polar_utils) (class in gfn.gym.helpers.box_utils) dists (gfn.utils.distributions.GraphActionDistribution attribute) diverse_replay_buffer (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) drain_pending_score() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) dt (gfn.estimators.PinnedBrownianMotionBackward attribute) (gfn.estimators.PinnedBrownianMotionForward attribute) (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.gym.diffusion_sampling.DiffusionSampling attribute) DUMMY (gfn.actions.GraphActionType attribute) dummy_action (gfn.actions.Actions attribute), [1] (gfn.actions.GraphActions attribute) (gfn.env.DiscreteEnv attribute) (gfn.env.Env attribute), [1] (gfn.env.GraphEnv attribute) dynamo_disable() (in module gfn.gflownet.mle)

E

edge_class (gfn.actions.GraphActions property) EDGE_CLASS_KEY (gfn.actions.GraphActions attribute), [1] edge_class_mlp (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) edge_embedding (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) edge_index (gfn.actions.GraphActions property) edge_index_action_to_src_dst() (gfn.actions.GraphActions class method) EDGE_INDEX_KEY (gfn.actions.GraphActions attribute), [1] edge_mlp (gfn.utils.modules.GraphEdgeActionMLP attribute) EdgePrior (class in gfn.gym.helpers.bayesian_structure.priors) edges_dim (gfn.utils.modules.GraphActionUniform attribute) (gfn.utils.modules.GraphEdgeActionMLP property) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNN property) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 property) (tutorials.examples.train_bayesian_structure.DAGEdgeActionMLP property) elapsed (gfn.utils.common.Timer attribute) (performance_tuning.BenchmarkResult attribute) embed_dim (gfn.utils.modules._AutoregressiveTransformerBlock attribute) embedding (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) (preprocessor.CodonSequencePreprocessor attribute) (tutorials.examples.train_chip_design.ChipDesignPreprocessor attribute) (tutorials.examples.train_chip_design_medium.ChipDesignPreprocessor attribute) embedding_dim (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) (gfn.utils.modules.SinusoidalPositionalEmbedding attribute) (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) (preprocessor.CodonSequencePreprocessor attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) (tutorials.examples.train_chip_design.ChipDesignPreprocessor attribute) (tutorials.examples.train_chip_design_medium.ChipDesignPreprocessor attribute) embedding_dropout (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) enabled (gfn.utils.common.Timer attribute) energy (gfn.gym.discrete_ebm.DiscreteEBM attribute), [1] (gfn.gym.DiscreteEBM attribute), [1] energy() (MFE_calculator.RNASolutionBase method) EnergyFunction (class in gfn.gym.discrete_ebm) ensure_same_device() (in module gfn.utils.common) EnumPreprocessor (class in gfn.preprocessors) env module Env (class in gfn.env) env (gfn.containers.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.ReplayBuffer attribute), [1] (gfn.containers.replay_buffer.TerminatingStateBuffer attribute) (gfn.containers.ReplayBuffer attribute), [1] (gfn.containers.states_container.StatesContainer attribute), [1] (gfn.containers.StatesContainer attribute), [1] (gfn.containers.TerminatingStateBuffer attribute) (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] (tutorials.examples.train_with_compile.TrainingComponents attribute) env() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_cartesian_utils.TestBoxCartesianEndToEnd method) (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianEndToEnd method) (test_box_polar_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) env_name (performance_tuning.BenchmarkConfig attribute) (performance_tuning.BenchmarkResult attribute) ENVIRONMENT_BENCHMARKS (in module tutorials.examples.train_with_compile) EnvironmentBenchmark (class in tutorials.examples.train_with_compile) EPS_INDEX_CMP (in module gfn.constants) EPS_REWARD_CMP (in module gfn.constants) eps_val (tutorials.examples.train_graph_ring.RingReward attribute) (tutorials.examples.train_graph_triangle.TriangleReward attribute) epsilon (gfn.gym.box.BoxPolar attribute), [1] (gfn.gym.BoxPolar attribute), [1] (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute), [1] (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1]

ErdosRenyiPrior (class in gfn.gym.helpers.bayesian_structure.priors) estimate_jsd() (in module tutorials.examples.train_box) (in module tutorials.examples.train_box_legacy) estimated_dist() (in module tutorials.examples.train_bit_sequences) (in module tutorials.examples.train_bitsequence_recurrent) Estimator (class in gfn.estimators) estimator (gfn.samplers.LocalSearchSampler attribute) (gfn.samplers.Sampler attribute), [1] estimator_outputs (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] evaluate module evaluate() (in module evaluate) evaluate_conditional_sampling() (in module tutorials.examples.train_conditional) evaluate_density_metrics() (in module tutorials.examples.train_diffusion_sampler) evaluate_l1() (in module tutorials.examples.train_bitsequence_non_autoregressive) EXAMPLES_OUTPUTS (in module tutorials.examples) EXIT (gfn.actions.GraphActionType attribute) (gfn.containers.message.MessageType attribute) exit_action (bench_get_scores_all._DBDummyActions attribute) (bench_get_scores_all._ModDBDummyActions attribute) (gfn.actions.Actions attribute), [1] (gfn.actions.GraphActions attribute) (gfn.env.DiscreteEnv attribute) (gfn.env.Env attribute), [1] (gfn.env.GraphEnv attribute) (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1] (tutorials.examples.train_line.ScaledGaussianWithOptionalExit attribute) exit_action_index (env.CodonDesignEnv attribute) exit_counter (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) exit_logits_scaled (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) exit_mlp (gfn.utils.modules.GraphEdgeActionMLP attribute) exit_probability (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1] expected_edges() (in module gfn.gym.helpers.bayesian_structure.evaluation) expected_output_dim (gfn.estimators.ConditionalScalarEstimator property) (gfn.estimators.DiffusionPolicyEstimator property) (gfn.estimators.DiscreteGraphPolicyEstimator property) (gfn.estimators.DiscretePolicyEstimator property) (gfn.estimators.Estimator property) (gfn.estimators.PinnedBrownianMotionBackward property) (gfn.estimators.PinnedBrownianMotionForward property) (gfn.estimators.ScalarEstimator property) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator property) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator property) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator property) (gfn.gym.helpers.box_polar_utils.BoxPFEstimator property) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator property) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator property) (gfn.gym.helpers.box_utils.BoxPBEstimator property) (gfn.gym.helpers.box_utils.BoxPFEstimator property) (tutorials.examples.train_line.StepEstimator property) expected_shd() (in module gfn.gym.helpers.bayesian_structure.evaluation) exponent_caps (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) extend() (gfn.actions.Actions method) (gfn.containers.base.Container method) (gfn.containers.Container method) (gfn.containers.replay_buffer.Container method) (gfn.containers.states_container.StatesContainer method) (gfn.containers.StatesContainer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions method) (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.chip_design.ChipDesignStates method) (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) (gfn.utils.graphs.GeometricBatch method) extend_with_dummy_actions() (gfn.actions.Actions method) extract_attribute_from_comments() (in module gfn.gym.helpers.chip_design.utils) extract_blockages_from_file() (in module gfn.gym.helpers.chip_design.utils) extract_sizes_from_comments() (in module gfn.gym.helpers.chip_design.utils) extras (gfn.estimators.RolloutContext attribute)

F

FairPrior (class in gfn.gym.helpers.bayesian_structure.priors) fd_placement_schedule() (in module gfn.gym.helpers.chip_design.utils) features_embedding (gfn.utils.modules.GraphEdgeActionMLP attribute) ff_dropout (gfn.utils.modules._AutoregressiveTransformerBlock attribute) ff_hidden_dim (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) fill_value (gfn.utils.modules.UniformModule attribute), [1] filter_kwargs_for_callable() (in module gfn.utils.common) filter_shift (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) filter_width (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) final_module (gfn.estimators.ConditionalDiscretePolicyEstimator attribute), [1] (gfn.estimators.ConditionalScalarEstimator attribute) final_norm (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) fix_macros_by_regex() (in module gfn.gym.helpers.chip_design.utils) fix_port_coordinates() (in module gfn.gym.helpers.chip_design.utils) fixed_length (gfn.gym.set_addition.SetAddition attribute), [1] (gfn.gym.SetAddition attribute), [1] flatten() (gfn.gym.bitSequence.BitSequenceStates method) (gfn.states.DiscreteStates method) (gfn.states.States method) flow_estimator (tutorials.examples.train_hypergrid_gafn.TBGAFN attribute) flow_matching_loss() (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method) flow_parameters() (tutorials.examples.train_hypergrid_gafn.TBGAFN method) FLOW_VARIANTS (in module tutorials.examples.train_with_compile) FlowVariant (class in tutorials.examples.train_with_compile) FMGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.flow_matching) forward() (gfn.estimators.ConditionalDiscretePolicyEstimator method) (gfn.estimators.ConditionalScalarEstimator method) (gfn.estimators.DiffusionPolicyEstimator method) (gfn.estimators.Estimator method) (gfn.estimators.PinnedBrownianMotionBackward method) (gfn.estimators.PinnedBrownianMotionForward method) (gfn.estimators.RecurrentDiscretePolicyEstimator method) (gfn.estimators.ScalarEstimator method) (gfn.gym.discrete_ebm.EnergyFunction method) (gfn.gym.discrete_ebm.IsingModel method) (gfn.gym.helpers.box_cartesian_utils._BoxCartesianMLP method) (gfn.gym.helpers.box_polar_utils.BoxPBMLP method) (gfn.gym.helpers.box_polar_utils.BoxPFMLP method) (gfn.gym.helpers.box_polar_utils.BoxStateFlowModule method) (gfn.gym.helpers.box_utils.BoxPBMLP method) (gfn.gym.helpers.box_utils.BoxPFMLP method) (gfn.gym.helpers.box_utils.BoxStateFlowModule method) (gfn.utils.modules._AutoregressiveTransformerBlock method) (gfn.utils.modules.AutoregressiveDiscreteSequenceModel method) (gfn.utils.modules.DiffusionFixedBackwardModule method) (gfn.utils.modules.DiffusionPISGradNetBackward method) (gfn.utils.modules.DiffusionPISGradNetForward method) (gfn.utils.modules.DiffusionPISJointPolicy method) (gfn.utils.modules.DiffusionPISStateEncoding method) (gfn.utils.modules.DiffusionPISTimeEncoding method) (gfn.utils.modules.GraphActionGNN method) (gfn.utils.modules.GraphActionUniform method) (gfn.utils.modules.GraphEdgeActionMLP method) (gfn.utils.modules.GraphScalarMLP method) (gfn.utils.modules.LinearTransformer method) (gfn.utils.modules.MLP method) (gfn.utils.modules.RecurrentDiscreteSequenceModel method) (gfn.utils.modules.SinusoidalPositionalEmbedding method) (gfn.utils.modules.Tabular method) (gfn.utils.modules.TransformerDiscreteSequenceModel method) (gfn.utils.modules.UniformModule method) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNN method) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 method) (tutorials.examples.train_hypergrid_gafn.RND method) (tutorials.examples.train_line.GaussianStepMLP method)

forward_kwargs() (in module tutorials.examples.train_diffusion_rtb) forward_looking (gfn.gflownet.DBGFlowNet attribute), [1] (gfn.gflownet.detailed_balance.DBGFlowNet attribute), [1] (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute), [1] (gfn.gflownet.SubTBGFlowNet attribute), [1] forward_masks (gfn.states.DiscreteStates attribute) (gfn.states.DiscreteStates property) (gfn.states.GraphStates property) freeze() (gfn.gym.helpers.bayesian_structure.jsd.GraphCollection method) from_batch_shape() (gfn.states.States class method) from_edge_indices() (in module gfn.utils.graphs) from_tensor_dict() (gfn.actions.GraphActions class method) from_tensordict() (gfn.containers.base.Container class method) (gfn.containers.Container class method) (gfn.containers.states_container.StatesContainer class method) (gfn.containers.StatesContainer class method) (gfn.containers.Trajectories class method) (gfn.containers.trajectories.Trajectories class method) (gfn.containers.Transitions class method) (gfn.containers.transitions.Transitions class method) FullPosterior (class in gfn.gym.helpers.bayesian_structure.jsd) Funnel (class in gfn.gym.diffusion_sampling)

G

gather_distributed_data() (in module gfn.utils.distributed) GaussianStepMLP (class in tutorials.examples.train_line) generate_all_rings() (in module tutorials.examples.train_with_example_modes) GeometricBatch (class in gfn.utils.graphs) get_arrays() (MFE_calculator.RNASolutionBase2 method) get_available_cpus() (in module gfn.utils.common) get_bitwise_xor_presets() (in module gfn.gym.hypergrid) get_blockages_from_comments() (in module gfn.gym.helpers.chip_design.utils) get_children() (in module gfn.gym.helpers.bayesian_structure.jsd) get_conditional_multiscale_presets() (in module gfn.gym.hypergrid) get_corrupted_presets() (in module gfn.gym.hypergrid) get_cosine_presets() (in module gfn.gym.hypergrid) get_current_estimator_output() (gfn.estimators.PolicyMixin method) get_data() (in module gfn.gym.helpers.bayesian_structure.factories) get_debug_metrics() (in module tutorials.examples.train_diffusion_rtb) get_deceptive_presets() (in module gfn.gym.hypergrid) get_edge_indices() (in module gfn.utils.graphs) 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_exact_P_T() (in module tutorials.examples.train_hypergrid) get_exploration_std() (in module tutorials.examples.train_diffusion_rtb) get_full_posterior() (in module gfn.gym.helpers.bayesian_structure.jsd) get_geometric_within_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_gflownet_cache() (in module gfn.gym.helpers.bayesian_structure.jsd) get_gfn_exact_posterior() (in module gfn.gym.helpers.bayesian_structure.jsd) get_macro_orientations() (in module gfn.gym.helpers.chip_design.utils) get_markov_blanket() (in module gfn.gym.helpers.bayesian_structure.jsd) get_markov_blanket_graph() (in module gfn.gym.helpers.bayesian_structure.jsd) GET_METADATA (gfn.containers.message.MessageType attribute) get_metadata() (gfn.containers.replay_buffer_manager.ReplayBufferManager static method) get_min_and_max_angles() (gfn.gym.helpers.box_polar_utils.QuarterCircle method) (gfn.gym.helpers.box_utils.QuarterCircle method) get_modified_db_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_multiplicative_coprime_presets() (in module gfn.gym.hypergrid) get_node_locations() (in module gfn.gym.helpers.chip_design.utils) get_node_ordering_by_size() (in module gfn.gym.helpers.chip_design.utils) get_node_type_counts() (in module gfn.gym.helpers.chip_design.utils) get_node_xy_coordinates() (in module gfn.gym.helpers.chip_design.utils) get_ordered_node_indices() (in module gfn.gym.helpers.chip_design.utils) get_original_presets() (in module gfn.gym.hypergrid) get_pfs_and_pbs() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) GET_POPULATION_STATS (gfn.containers.message.MessageType attribute) get_prior() (in module gfn.gym.helpers.bayesian_structure.factories) get_rank() (in module gfn.utils.distributed) get_reward_presets() (in module gfn.gym.hypergrid) get_scorer() (in module gfn.gym.helpers.bayesian_structure.factories) get_scores() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet method) (gfn.gflownet.ModifiedDBGFlowNet method) (gfn.gflownet.RelativeTBBase method) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) (gfn.gflownet.trajectory_balance.RelativeTBBase method) (gfn.gflownet.TrajectoryBasedGFlowNet method) (tutorials.examples.train_hypergrid_gafn.TBGAFN method) get_sparse_presets() (in module gfn.gym.hypergrid) get_states_indices (gfn.preprocessors.EnumPreprocessor attribute), [1] (gfn.preprocessors.OneHotPreprocessor attribute), [1] 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) (gfn.gym.perfect_tree.PerfectBinaryTree method) (gfn.gym.PerfectBinaryTree method) (gfn.gym.set_addition.SetAddition method) (gfn.gym.SetAddition method) get_synonymous_indices() (in module utils) get_tb_contributions() (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) get_terminating_state_dist() (gfn.env.DiscreteEnv method) (in module gfn.utils.training) 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 tutorials.examples.train_box) (in module tutorials.examples.train_box_legacy) get_train_group() (gfn.utils.distributed.DistributedContext method) get_trajectory_pbs() (in module gfn.utils.prob_calculations) get_trajectory_pfs() (in module gfn.utils.prob_calculations) get_trajectory_pfs_and_pbs() (in module gfn.utils.prob_calculations) get_transition_pbs() (in module gfn.utils.prob_calculations) get_transition_pfs() (in module gfn.utils.prob_calculations) get_transition_pfs_and_pbs() (in module gfn.utils.prob_calculations) get_uniform_random_presets() (in module gfn.gym.hypergrid) get_valid_actions() (in module gfn.gym.helpers.bayesian_structure.jsd) get_world_size() (in module gfn.utils.distributed) getK() (MFE_calculator.RNASolutionBase method) getM() (MFE_calculator.RNASolutionBase method) GetStatesIndicesFn (in module gfn.preprocessors) getU() (MFE_calculator.RNASolutionBase method) getV() (MFE_calculator.RNASolutionBase method) GFlowNet (class in gfn.gflownet) (class in gfn.gflownet.base) gflownet (tutorials.examples.train_with_compile.TrainingComponents attribute) gfn module gfn.actions module gfn.constants module gfn.containers module gfn.containers.base module gfn.containers.message module gfn.containers.replay_buffer module gfn.containers.replay_buffer_manager module gfn.containers.states_container module gfn.containers.trajectories module gfn.containers.transitions module gfn.env module gfn.estimators module gfn.gflownet module gfn.gflownet.base module gfn.gflownet.detailed_balance module

gfn.gflownet.flow_matching module gfn.gflownet.losses module gfn.gflownet.mle module gfn.gflownet.sub_trajectory_balance module gfn.gflownet.trajectory_balance module gfn.gym module gfn.gym.bayesian_structure module gfn.gym.bitSequence module gfn.gym.bitSequenceNonAutoregressive module gfn.gym.box module gfn.gym.box_cartesian module gfn.gym.chip_design module gfn.gym.diffusion_sampling module gfn.gym.discrete_ebm module gfn.gym.graph_building module gfn.gym.helpers module gfn.gym.helpers.bayesian_structure module gfn.gym.helpers.bayesian_structure.evaluation module gfn.gym.helpers.bayesian_structure.factories module gfn.gym.helpers.bayesian_structure.graph module gfn.gym.helpers.bayesian_structure.jsd module gfn.gym.helpers.bayesian_structure.priors module gfn.gym.helpers.bayesian_structure.sampling module gfn.gym.helpers.bayesian_structure.scores module gfn.gym.helpers.box_cartesian_utils module gfn.gym.helpers.box_polar_utils module gfn.gym.helpers.box_utils module gfn.gym.helpers.chip_design module gfn.gym.helpers.chip_design.build_container module gfn.gym.helpers.chip_design.plc_client module gfn.gym.helpers.chip_design.utils module gfn.gym.helpers.diffusion_utils module gfn.gym.hypergrid module gfn.gym.line module gfn.gym.perfect_tree module gfn.gym.set_addition module gfn.modules module gfn.preprocessors module gfn.samplers module gfn.states module gfn.utils module gfn.utils.common module gfn.utils.compile module gfn.utils.distributed module gfn.utils.distributions module gfn.utils.graphs module gfn.utils.handlers module gfn.utils.modules module gfn.utils.prob_calculations module gfn.utils.training module gfn.utils.trust_pcl module gfn_clip (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute) GFN_FNS (in module tutorials.examples.train_conditional) gmm (gfn.gym.diffusion_sampling.Grid25GaussianMixture attribute) grad_log_reward() (gfn.gym.diffusion_sampling.BaseTarget method) grad_norm() (in module gfn.utils.training) graph_network (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) graph_states_share_storage() (in module gfn.utils.graphs) GraphActionDistribution (class in gfn.utils.distributions) GraphActionGNN (class in gfn.utils.modules) GraphActions (class in gfn.actions) GraphActionType (class in gfn.actions) GraphActionUniform (class in gfn.utils.modules) GraphBuilding (class in gfn.gym) (class in gfn.gym.graph_building) GraphBuildingOnEdges (class in gfn.gym) (class in gfn.gym.graph_building) GraphCollection (class in gfn.gym.helpers.bayesian_structure.jsd) GraphEdgeActionMLP (class in gfn.utils.modules) GraphEnv (class in gfn.env) graphs (gfn.gym.helpers.bayesian_structure.jsd.FullPosterior attribute) GraphScalarMLP (class in gfn.utils.modules) GraphStates (class in gfn.states) Grid25GaussianMixture (class in gfn.gym.diffusion_sampling) grid_locations_near() (in module gfn.gym.helpers.chip_design.utils) GridReward (class in gfn.gym.hypergrid) Group (in module gfn.utils.distributed) gru (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) gt_logz() (gfn.gym.diffusion_sampling.BaseTarget method) (gfn.gym.diffusion_sampling.Funnel method) (gfn.gym.diffusion_sampling.Grid25GaussianMixture method) (gfn.gym.diffusion_sampling.ManyWell method) (gfn.gym.diffusion_sampling.Posterior9of25GaussianMixture method) (gfn.gym.diffusion_sampling.SimpleGaussianMixture method) gt_xs (gfn.gym.diffusion_sampling.BaseTarget attribute) gt_xs_log_rewards (gfn.gym.diffusion_sampling.BaseTarget attribute)

H

H (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute) (gfn.gym.BitSequencePlus attribute) (gfn.gym.NonAutoregressiveBitSequence attribute) HalfSquaredLoss (class in gfn.gflownet) (class in gfn.gflownet.losses) hamming_distance() (gfn.gym.BitSequence static method) (gfn.gym.bitSequence.BitSequence static method) harmonics_dim (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] has_conditions (gfn.states.States property) has_conditions_exception_handler() (in module gfn.utils.handlers) has_log_probs (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.containers.base.Container property) (gfn.containers.Container property) hash_graph() (in module gfn.utils.graphs) head (gfn.utils.modules.GraphScalarMLP attribute)

head_bit_range (gfn.gym.hypergrid.BitwiseXORReward attribute) head_check_count (gfn.gym.hypergrid.BitwiseXORReward attribute) head_dim (gfn.utils.modules._AutoregressiveTransformerBlock attribute) (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) head_seed (gfn.gym.hypergrid.BitwiseXORReward attribute) (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) head_weight (gfn.gym.hypergrid.BitwiseXORReward attribute) height (gfn.gym.HyperGrid attribute), [1] (gfn.gym.hypergrid.GridReward attribute) (gfn.gym.hypergrid.HyperGrid attribute), [1] (gfn.preprocessors.KHotPreprocessor attribute), [1] hidden_dim (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] (gfn.utils.modules.GraphEdgeActionMLP attribute) hidden_size (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) HyperGrid (class in gfn.gym) (class in gfn.gym.hypergrid) HYPERGRID_DEFAULTS (in module tutorials.examples.train_with_compile)

I

IdentityPreprocessor (class in gfn.preprocessors) idx_at_final_backward_step (tutorials.examples.train_line.ScaledGaussianWithOptionalExit attribute) idx_at_final_forward_step (tutorials.examples.train_line.ScaledGaussianWithOptionalExit attribute) idx_is_initial (gfn.gym.helpers.box_polar_utils.DistributionWrapper attribute), [1] (gfn.gym.helpers.box_utils.DistributionWrapper attribute), [1] idx_not_initial (gfn.gym.helpers.box_polar_utils.DistributionWrapper attribute), [1] (gfn.gym.helpers.box_utils.DistributionWrapper attribute), [1] idx_to_codon (env.CodonDesignEnv attribute) IDX_TO_CODON (in module utils) in_features (gfn.utils.modules.NoisyLinear attribute) init_carry() (gfn.estimators.RecurrentDiscretePolicyEstimator method) (gfn.utils.modules.AutoregressiveDiscreteSequenceModel method) (gfn.utils.modules.RecurrentDiscreteSequenceModel method) (gfn.utils.modules.TransformerDiscreteSequenceModel method) init_context() (gfn.estimators.PolicyEstimatorProtocol method) (gfn.estimators.PolicyMixin method) (gfn.estimators.RecurrentPolicyMixin method) init_env() (in module tutorials.examples.train_graph_ring) (in module tutorials.examples.train_graph_triangle) init_forward_masks() (gfn.states.DiscreteStates method) init_gflownet() (in module tutorials.examples.train_graph_ring) (in module tutorials.examples.train_graph_triangle) init_value (gfn.gym.Line attribute), [1] (gfn.gym.line.Line attribute), [1] initialize() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) initialize_distributed_compute() (in module gfn.utils.distributed) initialize_distributed_compute_mpi4py() (in module gfn.utils.distributed) input_dim (gfn.utils.modules.DiffusionFixedBackwardModule attribute), [1] (gfn.utils.modules.DiffusionPISGradNetForward attribute) (gfn.utils.modules.GraphActionGNN property) (gfn.utils.modules.GraphActionUniform attribute) (gfn.utils.modules.GraphEdgeActionMLP property) (gfn.utils.modules.GraphScalarMLP attribute) (gfn.utils.modules.MLP property) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 property) (tutorials.examples.train_line.GaussianStepMLP attribute) int_as_base() (in module MFE_utils) integers_to_binary() (gfn.gym.BitSequence static method) (gfn.gym.bitSequence.BitSequence static method) intermediary_states (gfn.containers.states_container.StatesContainer property) (gfn.containers.StatesContainer property) inverse_transition_table (gfn.gym.perfect_tree.PerfectBinaryTree attribute) (gfn.gym.PerfectBinaryTree attribute) irna (MFE_calculator.RNASolutionBase attribute) is_action_valid() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.box.BoxPolar method) (gfn.gym.box_cartesian.BoxCartesian method) (gfn.gym.BoxCartesian method) (gfn.gym.BoxPolar method) (gfn.gym.diffusion_sampling.DiffusionSampling method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.graph_building.GraphBuildingOnEdges method) (gfn.gym.GraphBuilding method) (gfn.gym.GraphBuildingOnEdges method) (gfn.gym.Line method) (gfn.gym.line.Line method) is_backward (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] (gfn.estimators.ConditionalDiscretePolicyEstimator attribute) (gfn.estimators.ConditionalScalarEstimator attribute) (gfn.estimators.DiscreteGraphPolicyEstimator attribute) (gfn.estimators.DiscretePolicyEstimator attribute) (gfn.estimators.Estimator attribute), [1] (gfn.estimators.LogitBasedEstimator attribute) (gfn.estimators.RecurrentDiscretePolicyEstimator attribute) (gfn.estimators.ScalarEstimator attribute) (gfn.utils.distributions.GraphActionDistribution attribute) (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute)

is_buffer_rank() (gfn.utils.distributed.DistributedContext method) (gfn.utils.distributed.DistributedContextMPI4Py method) is_callable_exception_handler() (in module gfn.utils.handlers) is_complete() (gfn.utils.distributed.AsyncSendHandle method) is_conditional (gfn.env.Env attribute), [1] (gfn.gym.ConditionalHyperGrid attribute) (gfn.gym.hypergrid.ConditionalHyperGrid attribute) is_coordinator_rank() (gfn.utils.distributed.DistributedContext method) is_directed (gfn.env.GraphEnv attribute), [1] (gfn.gym.graph_building.GraphBuilding attribute) (gfn.gym.GraphBuilding attribute) (gfn.states.GraphStates attribute), [1] (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) (gfn.utils.modules.GraphScalarMLP attribute) is_discrete (gfn.env.DiscreteEnv attribute), [1] (gfn.env.Env attribute), [1] is_dummy (gfn.actions.Actions property) (gfn.actions.GraphActions property) is_exit (bench_get_scores_all._DBDummyActions attribute) (bench_get_scores_all._ModDBDummyActions attribute) (gfn.actions.Actions property) (gfn.actions.GraphActions property) is_exit_actions() (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) is_frozen() (gfn.gym.helpers.bayesian_structure.jsd.GraphCollection method) is_initial_state (gfn.states.GraphStates property) (gfn.states.States property) is_int_dtype() (in module gfn.utils.common) is_running (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) is_s0 (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) is_sink_state (bench_get_scores_all._DBDummyStates attribute) (bench_get_scores_all._ModDBDummyStates attribute) (gfn.states.GraphStates property) (gfn.states.States property) is_terminal() (env.CodonDesignEnv method) is_terminating (bench_get_scores_all._DBDummyTransitions attribute) (gfn.containers.states_container.StatesContainer attribute), [1] (gfn.containers.StatesContainer attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] is_training_rank() (gfn.utils.distributed.DistributedContext method) (gfn.utils.distributed.DistributedContextMPI4Py method) is_vectorized (gfn.estimators.PolicyEstimatorProtocol attribute) (gfn.estimators.PolicyMixin property) (gfn.estimators.RecurrentPolicyMixin property) isend() (in module gfn.utils.distributed) IsingModel (class in gfn.gym.discrete_ebm) IsotropicGaussian (class in gfn.utils.distributions)

J

J (gfn.gym.discrete_ebm.IsingModel attribute), [1] jensen_shannon_divergence() (in module gfn.gym.helpers.bayesian_structure.jsd) joint_layers (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1]

joint_model (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1]

K

K (MFE_calculator.RNASolutionBase attribute) k_proj (gfn.utils.modules._AutoregressiveTransformerBlock attribute) k_select (gfn.gym.hypergrid.BitwiseXORReward attribute) key (gfn.utils.common.Timer attribute) (tutorials.examples.train_with_compile.CompileMode attribute) (tutorials.examples.train_with_compile.EnvironmentBenchmark attribute) (tutorials.examples.train_with_compile.FlowVariant attribute)

key_names (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) KHotPreprocessor (class in gfn.preprocessors) kwargs (gfn.gym.hypergrid.GridReward attribute)

L

label (performance_tuning.BenchmarkConfig property) (tutorials.examples.train_with_compile.CompileMode attribute) (tutorials.examples.train_with_compile.EnvironmentBenchmark attribute) (tutorials.examples.train_with_compile.FlowVariant attribute) lamda (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute), [1] (gfn.gflownet.SubTBGFlowNet attribute), [1] LARGE_MODEL_FIELDS (in module tutorials.examples.train_with_compile) layers (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) lazy_sort (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) lcm() (in module gfn.gym.hypergrid) lcm_multiple() (in module gfn.gym.hypergrid) learn_variance (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute) legalize_placement() (in module gfn.gym.helpers.chip_design.utils) length (gfn.gym.bitSequence.BitSequenceStates attribute), [1] Line (class in gfn.gym) (class in gfn.gym.line) linear (gfn.gym.discrete_ebm.IsingModel attribute) linear1 (gfn.utils.modules._AutoregressiveTransformerBlock attribute) linear2 (gfn.utils.modules._AutoregressiveTransformerBlock attribute) LinearTransformer (class in gfn.utils.modules) LinexLoss (class in gfn.gflownet) (class in gfn.gflownet.losses) list_available_targets() (gfn.gym.diffusion_sampling.DiffusionSampling class method) lmin (MFE_calculator.RNAFolderBase attribute) load() (gfn.containers.base.Container class method) (gfn.containers.Container class method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) load_config() (in module CAI_calculator) (in module utils) loc (gfn.utils.distributions.IsotropicGaussian attribute), [1] local_score() (gfn.gym.helpers.bayesian_structure.scores.BGeScore method) local_search() (gfn.samplers.LocalSearchSampler method) LocalSearchSampler (class in gfn.samplers) locs (gfn.gym.diffusion_sampling.Grid25GaussianMixture attribute) log_gamma_term (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) log_partition() (gfn.env.Env method) (gfn.gym.box.BoxPolar method) (gfn.gym.BoxPolar method) (gfn.gym.ConditionalHyperGrid method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.ConditionalHyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.Line method) (gfn.gym.line.Line method) log_prior (gfn.gym.helpers.bayesian_structure.priors.BasePrior property) (gfn.gym.helpers.bayesian_structure.priors.EdgePrior property) (gfn.gym.helpers.bayesian_structure.priors.ErdosRenyiPrior property) (gfn.gym.helpers.bayesian_structure.priors.FairPrior property) (gfn.gym.helpers.bayesian_structure.priors.UniformPrior property) log_prob() (bench_get_scores_all._ModDBFakeDist method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution method) (gfn.gym.helpers.box_polar_utils.DistributionWrapper method) (gfn.gym.helpers.box_polar_utils.QuarterCircle method) (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit method) (gfn.gym.helpers.box_polar_utils.QuarterDisk method) (gfn.gym.helpers.box_utils.BoxCartesianDistribution method) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution 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.GraphActionDistribution method) (gfn.utils.distributions.IsotropicGaussian method) (gfn.utils.distributions.UnsqueezedCategorical method) (tutorials.examples.train_line.ScaledGaussianWithOptionalExit method) log_probas (gfn.gym.helpers.bayesian_structure.jsd.FullPosterior attribute) log_probs (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] log_probs() (gfn.estimators.PolicyEstimatorProtocol method) (gfn.estimators.PolicyMixin method) log_reward() (bench_get_scores_all._DBDummyEnv method) (gfn.env.Env method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) (gfn.gym.diffusion_sampling.BaseTarget method) (gfn.gym.diffusion_sampling.DiffusionSampling method) (gfn.gym.diffusion_sampling.Funnel method) (gfn.gym.diffusion_sampling.Grid25GaussianMixture method) (gfn.gym.diffusion_sampling.ManyWell method) (gfn.gym.diffusion_sampling.Posterior9of25GaussianMixture method) (gfn.gym.diffusion_sampling.SimpleGaussianMixture method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.Line method) (gfn.gym.line.Line method) (gfn.gym.NonAutoregressiveBitSequence method) log_reward_clip_min (gfn.gflownet.base.GFlowNet attribute) (gfn.gflownet.base.PFBasedGFlowNet attribute), [1] (gfn.gflownet.base.TrajectoryBasedGFlowNet attribute) (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute) (gfn.gflownet.GFlowNet attribute) (gfn.gflownet.LogPartitionVarianceGFlowNet attribute) (gfn.gflownet.PFBasedGFlowNet attribute), [1] (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) (gfn.gflownet.TrajectoryBasedGFlowNet attribute)

log_rewards (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._TBTrajectoriesStub property) (gfn.containers.base.Container property) (gfn.containers.Container property) (gfn.containers.replay_buffer.Container property) (gfn.containers.states_container.StatesContainer property) (gfn.containers.StatesContainer property) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) log_var_range (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute) log_weights (gfn.gym.helpers.box_cartesian_utils._BetaMixtureMixin attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) logdet() (in module gfn.gym.helpers.bayesian_structure.scores) logF (gfn.gflownet.DBGFlowNet attribute), [1] (gfn.gflownet.detailed_balance.DBGFlowNet attribute), [1] (gfn.gflownet.flow_matching.FMGFlowNet attribute), [1] (gfn.gflownet.FMGFlowNet attribute), [1] (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute), [1] (gfn.gflownet.SubTBGFlowNet attribute), [1] logF_named_parameters() (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) logF_parameters() (gfn.gflownet.DBGFlowNet method) (gfn.gflownet.detailed_balance.DBGFlowNet method) (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet method) (gfn.gflownet.SubTBGFlowNet method) logger (in module gfn.gym.diffusion_sampling) (in module gfn.gym.helpers.chip_design.plc_client) (in module gfn.gym.hypergrid) (in module gfn.modules) (in module gfn.states) (in module gfn.utils.common) (in module gfn.utils.compile) (in module gfn.utils.distributed) (in module gfn.utils.handlers) (in module gfn.utils.modules) (in module gfn.utils.training) (in module tutorials.examples.train_hypergrid) LogitBasedEstimator (class in gfn.estimators) 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.RelativeTrajectoryBalanceGFlowNet attribute) (gfn.gflownet.TBGFlowNet attribute), [1] (gfn.gflownet.trajectory_balance.RelativeTrajectoryBalanceGFlowNet attribute) (gfn.gflownet.trajectory_balance.TBGFlowNet attribute), [1] logz_named_parameters() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) logz_parameters() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) logZ_value (gfn.gym.helpers.box_polar_utils.BoxStateFlowModule attribute), [1] (gfn.gym.helpers.box_utils.BoxStateFlowModule attribute), [1] 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.mle.MLEDiffusion method) (gfn.gflownet.ModifiedDBGFlowNet method) (gfn.gflownet.RelativeLogPartitionVarianceGFlowNet method) (gfn.gflownet.RelativeTrajectoryBalanceGFlowNet 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.RelativeLogPartitionVarianceGFlowNet method) (gfn.gflownet.trajectory_balance.RelativeTrajectoryBalanceGFlowNet method) (gfn.gflownet.trajectory_balance.TBGFlowNet method) (gfn.gflownet.trajectory_balance.TrustPCLGFlowNet method) (gfn.gflownet.TrustPCLGFlowNet method) (tutorials.examples.train_hypergrid_gafn.TBGAFN method) loss_fn (gfn.gflownet.base.GFlowNet attribute) (gfn.gflownet.GFlowNet attribute) loss_from_trajectories() (gfn.gflownet.base.GFlowNet method) (gfn.gflownet.GFlowNet method) LOSS_LINE_ALPHA (in module tutorials.examples.train_with_compile) loss_name (performance_tuning.BenchmarkConfig attribute) loss_reduce() (in module gfn.gflownet.base) loss_scale (tutorials.examples.train_hypergrid_gafn.RND attribute) LR (in module performance_tuning) lr_grad_ratio() (in module gfn.utils.training) LR_LOGF (in module performance_tuning) LR_LOGZ (in module performance_tuning) lstm (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute)

M

M (MFE_calculator.RNASolutionBase attribute) main module main() (in module bench_get_scores_all) (in module main) (in module performance_tuning) (in module tutorials.examples.train_bayesian_structure) (in module tutorials.examples.train_bit_sequences) (in module tutorials.examples.train_bitsequence_non_autoregressive) (in module tutorials.examples.train_bitsequence_recurrent) (in module tutorials.examples.train_box) (in module tutorials.examples.train_box_legacy) (in module tutorials.examples.train_chip_design) (in module tutorials.examples.train_chip_design_medium) (in module tutorials.examples.train_conditional) (in module tutorials.examples.train_diffusion_rtb) (in module tutorials.examples.train_diffusion_sampler) (in module tutorials.examples.train_discreteebm) (in module tutorials.examples.train_graph_ring) (in module tutorials.examples.train_graph_triangle) (in module tutorials.examples.train_hypergrid) (in module tutorials.examples.train_hypergrid_buffer) (in module tutorials.examples.train_hypergrid_exploration_examples) (in module tutorials.examples.train_hypergrid_gafn) (in module tutorials.examples.train_hypergrid_local_search) (in module tutorials.examples.train_hypergrid_simple) (in module tutorials.examples.train_ising) (in module tutorials.examples.train_line) (in module tutorials.examples.train_with_compile) (in module tutorials.examples.train_with_example_modes) make_actions_class() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.env.GraphEnv method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.GraphBuilding method) make_blockage_text() (in module gfn.gym.helpers.chip_design.utils) make_dataloader_seed_fns() (in module gfn.utils.common) make_dummy_actions() (gfn.actions.Actions class method) (gfn.actions.GraphActions class method) make_exit_actions() (gfn.actions.Actions class method) (gfn.actions.GraphActions class method) make_initial_states() (gfn.states.GraphStates class method) (gfn.states.States class method) make_modes_set() (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) make_random_states (gfn.states.States attribute), [1] make_random_states() (gfn.env.Env method) (gfn.env.GraphEnv method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.box.BoxPolar method) (gfn.gym.BoxPolar method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.graph_building.GraphBuildingOnEdges method) (gfn.gym.GraphBuilding method) (gfn.gym.GraphBuildingOnEdges method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.NonAutoregressiveBitSequence method) make_random_states_tensor() (gfn.gym.bayesian_structure.BayesianStructure method) make_sink_states() (gfn.states.GraphStates class method) (gfn.states.States class method) make_sink_states_tensor() (env.CodonDesignEnv static method) make_states_class() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.env.GraphEnv method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequence.BitSequencePlus method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.BitSequencePlus method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) (gfn.gym.diffusion_sampling.DiffusionSampling method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.graph_building.GraphBuildingOnEdges method) (gfn.gym.GraphBuilding method) (gfn.gym.GraphBuildingOnEdges method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.NonAutoregressiveBitSequence method) (gfn.gym.perfect_tree.PerfectBinaryTree method) (gfn.gym.PerfectBinaryTree method) (gfn.gym.set_addition.SetAddition method) (gfn.gym.SetAddition method) make_step_fn() (in module performance_tuning) ManyWell (class in gfn.gym.diffusion_sampling) mapping (gfn.gym.helpers.bayesian_structure.jsd.GraphCollection attribute) markov (gfn.gym.helpers.bayesian_structure.jsd.FullPosterior attribute) MaskTensor (in module gfn.gflownet.sub_trajectory_balance) max_angles (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] MAX_BULGE (MFE_calculator.RNAFolderBase attribute) max_concentration (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxPFEstimator attribute), [1] MAX_HAIRPIN (MFE_calculator.RNAFolderBase attribute) MAX_INTERNAL (MFE_calculator.RNAFolderBase attribute) max_items (gfn.gym.set_addition.SetAddition attribute) (gfn.gym.SetAddition attribute) max_length (bench_get_scores_all._SubTBDummyTrajectories attribute) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) max_nodes (gfn.gym.graph_building.GraphBuilding attribute) (gfn.gym.GraphBuilding attribute) (gfn.states.GraphStates attribute) max_position_embeddings (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) max_range (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) MAX_RETRY (gfn.gym.helpers.chip_design.plc_client.PlacementCost attribute) max_traj_len (gfn.gym.set_addition.SetAddition attribute) (gfn.gym.SetAddition attribute) maybe_compile() (in module performance_tuning) maybe_compile_estimators() (in module tutorials.examples.train_with_compile) mean (gfn.gym.chip_design.CostStats attribute) mean_iter_ms (performance_tuning.BenchmarkResult attribute) mean_obs (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) means (gfn.gym.diffusion_sampling.ManyWell attribute) MEDIUM_INIT_PLACEMENT (in module gfn.gym.helpers.chip_design) MEDIUM_NETLIST_FILE (in module gfn.gym.helpers.chip_design) Message (class in gfn.containers.message) message_data (gfn.containers.message.Message attribute) message_type (gfn.containers.message.Message attribute) MessageType (class in gfn.containers.message) METADATA_TAG (in module gfn.containers.replay_buffer_manager) MFE_calculator module MFE_utils module min (gfn.gym.chip_design.CostStats attribute) min_angles (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] min_concentration (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxPFEstimator attribute), [1] min_incr (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute)

mixture (gfn.gym.Line attribute), [1] (gfn.gym.line.Line attribute), [1] mixture_logits (gfn.gym.helpers.box_polar_utils.QuarterDisk attribute), [1] (gfn.gym.helpers.box_utils.QuarterDisk attribute), [1] MLEDiffusion (class in gfn.gflownet.mle) MLP (class in gfn.utils.modules) mlp (gfn.utils.modules.GraphEdgeActionMLP attribute) mode_mask() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) mode_prob (gfn.gym.hypergrid.UniformRandomReward attribute) MODE_REPORT (gfn.containers.message.MessageType attribute) mode_stats (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) mode_stats_samples (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) mode_threshold() (gfn.gym.hypergrid.ConditionalMultiScaleReward method) (gfn.gym.hypergrid.CorruptedReward method) mode_tier() (gfn.gym.hypergrid.ConditionalMultiScaleReward method) model (gfn.utils.modules.DiffusionPISJointPolicy attribute) modes (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute), [1] (gfn.gym.BitSequencePlus attribute) (gfn.gym.NonAutoregressiveBitSequence attribute), [1] modes_found() (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) ModifiedDBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.detailed_balance) module bench_get_scores_all CAI_calculator env evaluate gfn gfn.actions gfn.constants gfn.containers gfn.containers.base gfn.containers.message gfn.containers.replay_buffer gfn.containers.replay_buffer_manager gfn.containers.states_container gfn.containers.trajectories gfn.containers.transitions gfn.env gfn.estimators gfn.gflownet gfn.gflownet.base gfn.gflownet.detailed_balance gfn.gflownet.flow_matching gfn.gflownet.losses gfn.gflownet.mle gfn.gflownet.sub_trajectory_balance gfn.gflownet.trajectory_balance gfn.gym gfn.gym.bayesian_structure gfn.gym.bitSequence gfn.gym.bitSequenceNonAutoregressive gfn.gym.box gfn.gym.box_cartesian gfn.gym.chip_design gfn.gym.diffusion_sampling gfn.gym.discrete_ebm gfn.gym.graph_building gfn.gym.helpers gfn.gym.helpers.bayesian_structure gfn.gym.helpers.bayesian_structure.evaluation gfn.gym.helpers.bayesian_structure.factories gfn.gym.helpers.bayesian_structure.graph gfn.gym.helpers.bayesian_structure.jsd gfn.gym.helpers.bayesian_structure.priors gfn.gym.helpers.bayesian_structure.sampling gfn.gym.helpers.bayesian_structure.scores gfn.gym.helpers.box_cartesian_utils gfn.gym.helpers.box_polar_utils gfn.gym.helpers.box_utils gfn.gym.helpers.chip_design gfn.gym.helpers.chip_design.build_container gfn.gym.helpers.chip_design.plc_client gfn.gym.helpers.chip_design.utils gfn.gym.helpers.diffusion_utils gfn.gym.hypergrid gfn.gym.line gfn.gym.perfect_tree gfn.gym.set_addition gfn.modules gfn.preprocessors gfn.samplers gfn.states gfn.utils gfn.utils.common gfn.utils.compile gfn.utils.distributed gfn.utils.distributions gfn.utils.graphs gfn.utils.handlers gfn.utils.modules gfn.utils.prob_calculations gfn.utils.training gfn.utils.trust_pcl main MFE_calculator MFE_utils performance_tuning preprocessor test_box_cartesian_utils test_box_polar_utils tutorials tutorials.examples tutorials.examples.train_bayesian_structure tutorials.examples.train_bit_sequences tutorials.examples.train_bitsequence_non_autoregressive tutorials.examples.train_bitsequence_recurrent tutorials.examples.train_box tutorials.examples.train_box_legacy tutorials.examples.train_chip_design tutorials.examples.train_chip_design_medium tutorials.examples.train_conditional tutorials.examples.train_diffusion_rtb tutorials.examples.train_diffusion_sampler tutorials.examples.train_discreteebm tutorials.examples.train_graph_ring tutorials.examples.train_graph_triangle tutorials.examples.train_hypergrid tutorials.examples.train_hypergrid_buffer tutorials.examples.train_hypergrid_exploration_examples tutorials.examples.train_hypergrid_gafn tutorials.examples.train_hypergrid_local_search tutorials.examples.train_hypergrid_simple tutorials.examples.train_ising tutorials.examples.train_line tutorials.examples.train_with_compile tutorials.examples.train_with_example_modes utils module (gfn.estimators.ConditionalDiscretePolicyEstimator attribute) (gfn.estimators.ConditionalLogZEstimator attribute) (gfn.estimators.ConditionalScalarEstimator attribute) (gfn.estimators.DiscreteGraphPolicyEstimator attribute) (gfn.estimators.DiscretePolicyEstimator attribute) (gfn.estimators.Estimator attribute), [1] (gfn.estimators.LogitBasedEstimator attribute) (gfn.estimators.RecurrentDiscretePolicyEstimator attribute) (gfn.estimators.ScalarEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxPBEstimator attribute) (gfn.utils.modules.LinearTransformer attribute) MultiplicativeCoprimeReward (class in gfn.gym.hypergrid) mus (gfn.gym.Line attribute), [1] (gfn.gym.line.Line attribute), [1] my_rank (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute)

N

n_actions (env.CodonDesignEnv attribute) (gfn.env.DiscreteEnv attribute), [1] (gfn.estimators.ConditionalDiscretePolicyEstimator attribute), [1] (gfn.estimators.DiscretePolicyEstimator attribute), [1] (gfn.estimators.RecurrentDiscretePolicyEstimator attribute) (gfn.gym.bayesian_structure.BayesianStructure attribute) (gfn.gym.BitSequence attribute) (gfn.gym.bitSequence.BitSequence attribute) (gfn.gym.chip_design.ChipDesignStates attribute) (gfn.gym.perfect_tree.PerfectBinaryTree attribute), [1] (gfn.gym.PerfectBinaryTree attribute), [1] (gfn.states.DiscreteStates attribute), [1] N_CODONS (in module utils) n_components (gfn.gym.helpers.box_cartesian_utils._BoxCartesianMLP attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPBMLP attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPFMLP attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterDisk attribute), [1] (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxPBEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxPBMLP attribute), [1] (gfn.gym.helpers.box_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxPFMLP attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterDisk attribute), [1] n_components_s0 (gfn.gym.helpers.box_polar_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_polar_utils.BoxPFMLP attribute), [1] (gfn.gym.helpers.box_utils.BoxPFEstimator attribute), [1] (gfn.gym.helpers.box_utils.BoxPFMLP attribute), [1] n_dim (gfn.gym.helpers.box_cartesian_utils._BoxCartesianMLP attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) n_grid_cells (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) n_items (gfn.gym.set_addition.SetAddition attribute), [1] (gfn.gym.SetAddition attribute), [1] n_iters (performance_tuning.BenchmarkSettings attribute) n_macros (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) (tutorials.examples.train_chip_design.ChipDesignPreprocessor attribute) (tutorials.examples.train_chip_design_medium.ChipDesignPreprocessor attribute) n_mode_states (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) n_modes (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute) (gfn.gym.BitSequencePlus attribute) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) (gfn.gym.NonAutoregressiveBitSequence attribute) n_modes_count (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute) (gfn.gym.NonAutoregressiveBitSequence attribute) n_nodes (gfn.gym.bayesian_structure.BayesianStructure attribute) (gfn.gym.graph_building.GraphBuildingOnEdges attribute), [1] (gfn.gym.GraphBuildingOnEdges attribute), [1] (gfn.gym.perfect_tree.PerfectBinaryTree attribute), [1] (gfn.gym.PerfectBinaryTree attribute), [1] (gfn.utils.modules.GraphEdgeActionMLP attribute) (gfn.utils.modules.GraphScalarMLP attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNN attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 property) n_possible_edges (gfn.gym.graph_building.GraphBuildingOnEdges attribute) (gfn.gym.GraphBuildingOnEdges attribute) n_rules (gfn.gym.hypergrid.BitwiseXORReward attribute) (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) n_sd (gfn.gym.Line attribute), [1] (gfn.gym.line.Line attribute), [1] n_states (gfn.env.DiscreteEnv property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircleWithExit attribute), [1] (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) n_states3 (gfn.gym.BitSequence property) (gfn.gym.bitSequence.BitSequence property) n_steps_per_trajectory (gfn.gym.Line attribute), [1] (gfn.gym.line.Line attribute), [1] (tutorials.examples.train_line.StepEstimator attribute)

n_terminating_states (gfn.env.DiscreteEnv property) (gfn.gym.BitSequence property) (gfn.gym.bitSequence.BitSequence property) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) (gfn.gym.NonAutoregressiveBitSequence property) n_trajectories (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) n_transitions (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) n_variance_outputs (gfn.estimators.PinnedBrownianMotionBackward attribute) (gfn.estimators.PinnedBrownianMotionForward attribute) n_words (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute), [1] (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequenceStates attribute), [1] (gfn.gym.NonAutoregressiveBitSequence attribute), [1] ndim (gfn.gym.discrete_ebm.DiscreteEBM attribute), [1] (gfn.gym.DiscreteEBM attribute), [1] (gfn.gym.HyperGrid attribute), [1] (gfn.gym.hypergrid.GridReward attribute) (gfn.gym.hypergrid.HyperGrid attribute), [1] (gfn.preprocessors.KHotPreprocessor attribute), [1] next_states (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyTransitions attribute) (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] no_conditions_exception_handler() (in module gfn.utils.handlers) node_class (gfn.actions.GraphActions property) NODE_CLASS_KEY (gfn.actions.GraphActions attribute), [1] node_class_mlp (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) node_embedding (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) node_index (gfn.actions.GraphActions property) NODE_INDEX_KEY (gfn.actions.GraphActions attribute) node_index_mlp (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) nodes_of_types() (in module gfn.gym.helpers.chip_design.utils) NoisyLinear (class in gfn.utils.modules) non_pairing (MFE_calculator.RNAFolderBase attribute) NonAutoregressiveBitSequence (class in gfn.gym) (class in gfn.gym.bitSequenceNonAutoregressive) NonAutoregressiveBitSequenceStates (class in gfn.gym.bitSequenceNonAutoregressive) NonValidActionsError (in module gfn.env) norm (gfn.utils.modules.GraphActionGNN attribute) norm() (gfn.gym.box.BoxPolar static method) (gfn.gym.BoxPolar static method) norm1 (gfn.utils.modules._AutoregressiveTransformerBlock attribute) norm2 (gfn.utils.modules._AutoregressiveTransformerBlock attribute) NormBasedDiversePrioritizedReplayBuffer (class in gfn.containers) (class in gfn.containers.replay_buffer) northeastern (gfn.gym.helpers.box_polar_utils.QuarterCircle attribute), [1] (gfn.gym.helpers.box_utils.QuarterCircle attribute), [1] notes (tutorials.examples.train_with_compile.TrainingComponents attribute) num_conv_layers (gfn.utils.modules.GraphActionGNN attribute) NUM_DAGS (in module gfn.gym.helpers.bayesian_structure.jsd) num_discretization_steps (gfn.estimators.PinnedBrownianMotionForward attribute) num_edge_classes (gfn.env.GraphEnv attribute), [1] (gfn.gym.graph_building.GraphBuilding attribute) (gfn.gym.GraphBuilding attribute) (gfn.states.GraphStates attribute), [1] (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphActionUniform attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) num_edges_per_node (gfn.gym.helpers.bayesian_structure.priors.ErdosRenyiPrior attribute) num_heads (gfn.utils.modules._AutoregressiveTransformerBlock attribute) (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) num_layers (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) num_levels (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) num_node_classes (gfn.env.GraphEnv attribute), [1] (gfn.gym.graph_building.GraphBuilding attribute) (gfn.gym.GraphBuilding attribute) (gfn.states.GraphStates attribute), [1] (gfn.utils.modules.GraphActionGNN attribute) (gfn.utils.modules.GraphActionUniform attribute) (gfn.utils.modules.GraphEdgeActionMLP attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) num_nodes (gfn.gym.helpers.bayesian_structure.scores.BaseScore attribute) (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) num_samples (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) num_steps (gfn.gflownet.mle.MLEDiffusion attribute) num_training_ranks (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) num_variables (gfn.gym.helpers.bayesian_structure.priors.BasePrior property) numerical_epsilon (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) nx_to_geometric_data() (in module gfn.gym.helpers.bayesian_structure.jsd)

O

OneHotPreprocessor (class in gfn.preprocessors) open_energy() (MFE_calculator.RNAFolderBase2 method) optimize_orientations() (in module gfn.gym.helpers.chip_design.utils) optimizer (tutorials.examples.train_with_compile.TrainingComponents attribute) ORIENTATIONS (in module gfn.gym.helpers.chip_design.utils) OriginalReward (class in gfn.gym.hypergrid) out_dim (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] out_features (gfn.utils.modules.NoisyLinear attribute) out_proj (gfn.utils.modules._AutoregressiveTransformerBlock attribute) output_dim (gfn.preprocessors.EnumPreprocessor attribute) (gfn.preprocessors.IdentityPreprocessor attribute) (gfn.preprocessors.KHotPreprocessor attribute), [1] (gfn.preprocessors.OneHotPreprocessor attribute), [1] (gfn.preprocessors.Preprocessor attribute), [1] (gfn.utils.modules.GraphActionUniform attribute) (gfn.utils.modules.GraphEdgeActionMLP property) (gfn.utils.modules.LinearTransformer attribute) (gfn.utils.modules.MLP property) (gfn.utils.modules.UniformModule attribute), [1] (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 property) (tutorials.examples.train_chip_design.ChipDesignPreprocessor attribute) (tutorials.examples.train_chip_design_medium.ChipDesignPreprocessor attribute) (tutorials.examples.train_line.GaussianStepMLP attribute)

output_path (performance_tuning.BenchmarkSettings attribute) output_projection (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) (gfn.utils.modules.TransformerDiscreteSequenceModel attribute)

P

p_norm_distance (gfn.containers.NormBasedDiversePrioritizedReplayBuffer attribute), [1] (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute), [1] pad_dim0_if_needed() (in module gfn.containers.trajectories) pad_dim0_with_sf() (gfn.gym.bitSequence.BitSequenceStates method) (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) pair_int_as_str() (in module MFE_utils) pair_int_as_tuple() (in module MFE_utils) param_norm() (in module gfn.utils.training) parity_checks (gfn.gym.hypergrid.BitwiseXORReward attribute) parse_args() (in module tutorials.examples.train_with_compile) parser (in module gfn.gym.helpers.chip_design.build_container) (in module main) (in module tutorials.examples.train_bayesian_structure) (in module tutorials.examples.train_bit_sequences) (in module tutorials.examples.train_bitsequence_non_autoregressive) (in module tutorials.examples.train_bitsequence_recurrent) (in module tutorials.examples.train_box) (in module tutorials.examples.train_box_legacy) (in module tutorials.examples.train_chip_design) (in module tutorials.examples.train_chip_design_medium) (in module tutorials.examples.train_conditional) (in module tutorials.examples.train_diffusion_sampler) (in module tutorials.examples.train_discreteebm) (in module tutorials.examples.train_graph_ring) (in module tutorials.examples.train_graph_triangle) (in module tutorials.examples.train_hypergrid) (in module tutorials.examples.train_hypergrid_buffer) (in module tutorials.examples.train_hypergrid_exploration_examples) (in module tutorials.examples.train_hypergrid_gafn) (in module tutorials.examples.train_hypergrid_local_search) (in module tutorials.examples.train_hypergrid_simple) (in module tutorials.examples.train_ising) (in module tutorials.examples.train_line) (in module tutorials.examples.train_with_example_modes) pb (gfn.gflownet.base.PFBasedGFlowNet attribute), [1] (gfn.gflownet.base.TrajectoryBasedGFlowNet attribute) (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet attribute) (gfn.gflownet.LogPartitionVarianceGFlowNet attribute) (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.gflownet.ModifiedDBGFlowNet attribute) (gfn.gflownet.PFBasedGFlowNet attribute), [1] (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) (gfn.gflownet.TrajectoryBasedGFlowNet attribute) pb_estimator() (test_box_cartesian_utils.TestBoxCartesianEndToEnd method) (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianEndToEnd method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) pb_module() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) pb_scale_range (gfn.estimators.PinnedBrownianMotionBackward attribute) (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.utils.modules.DiffusionPISGradNetBackward attribute) pe (gfn.utils.modules.SinusoidalPositionalEmbedding property) pending_container (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) per_step_decay() (in module tutorials.examples.train_with_example_modes) PerfectBinaryTree (class in gfn.gym) (class in gfn.gym.perfect_tree) performance_tuning module pf (gfn.gflownet.base.PFBasedGFlowNet attribute), [1] (gfn.gflownet.base.TrajectoryBasedGFlowNet attribute) (gfn.gflownet.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.DBGFlowNet attribute) (gfn.gflownet.detailed_balance.ModifiedDBGFlowNet attribute) (gfn.gflownet.LogPartitionVarianceGFlowNet attribute) (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.gflownet.ModifiedDBGFlowNet attribute) (gfn.gflownet.PFBasedGFlowNet attribute), [1] (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) (gfn.gflownet.TrajectoryBasedGFlowNet attribute) pf_estimator() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_cartesian_utils.TestBoxCartesianEndToEnd method) (test_box_polar_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianEndToEnd method)

pf_module() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) pf_pb_named_parameters() (gfn.gflownet.base.PFBasedGFlowNet method) (gfn.gflownet.PFBasedGFlowNet method) pf_pb_parameters() (gfn.gflownet.base.PFBasedGFlowNet method) (gfn.gflownet.PFBasedGFlowNet method) PFBasedGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.base) PFs0 (gfn.gym.helpers.box_polar_utils.BoxPFMLP attribute), [1] (gfn.gym.helpers.box_utils.BoxPFMLP attribute), [1] PI_2 (in module gfn.gym.helpers.box_polar_utils) PI_2_INV (in module gfn.gym.helpers.box_polar_utils) PinnedBrownianMotionBackward (class in gfn.estimators) PinnedBrownianMotionForward (class in gfn.estimators) place_near() (in module gfn.gym.helpers.chip_design.utils) PlacementCost (class in gfn.gym.helpers.chip_design.plc_client) plc (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) PLC_DOWNLOAD_URL (in module gfn.gym.helpers.chip_design) plot_benchmark() (in module tutorials.examples.train_with_compile) plot_border (gfn.gym.diffusion_sampling.BaseTarget attribute) plot_results() (in module performance_tuning) (in module tutorials.examples.train_hypergrid) plot_samples() (in module tutorials.examples.train_diffusion_rtb) plot_training_curves() (in module utils) plot_trajectories() (in module tutorials.examples.train_box) policy_std_max (tutorials.examples.train_line.GaussianStepMLP attribute) policy_std_min (tutorials.examples.train_line.GaussianStepMLP attribute) PolicyEstimatorProtocol (class in gfn.estimators) PolicyMixin (class in gfn.estimators) posterior (gfn.gym.diffusion_sampling.Posterior9of25GaussianMixture attribute) Posterior9of25GaussianMixture (class in gfn.gym.diffusion_sampling) posterior_estimate() (in module gfn.gym.helpers.bayesian_structure.evaluation) posterior_exact() (in module gfn.gym.helpers.bayesian_structure.jsd) precompute() (gfn.gym.chip_design.CostStats static method) PredictionsTensor (in module gfn.gflownet.sub_trajectory_balance) predictor_net (tutorials.examples.train_hypergrid_gafn.RND attribute) prepare_loss_fn() (in module tutorials.examples.train_with_compile) preprocess() (gfn.preprocessors.EnumPreprocessor method) (gfn.preprocessors.IdentityPreprocessor method) (gfn.preprocessors.KHotPreprocessor method) (gfn.preprocessors.OneHotPreprocessor method) (gfn.preprocessors.Preprocessor method) (preprocessor.CodonSequencePreprocessor method) (tutorials.examples.train_chip_design.ChipDesignPreprocessor method) (tutorials.examples.train_chip_design_medium.ChipDesignPreprocessor method) preprocessor module Preprocessor (class in gfn.preprocessors) preprocessor (gfn.estimators.ConditionalDiscretePolicyEstimator attribute) (gfn.estimators.ConditionalScalarEstimator attribute) (gfn.estimators.DiscreteGraphPolicyEstimator attribute) (gfn.estimators.DiscretePolicyEstimator attribute) (gfn.estimators.Estimator attribute), [1] (gfn.estimators.LogitBasedEstimator attribute) (gfn.estimators.RecurrentDiscretePolicyEstimator attribute) (gfn.estimators.ScalarEstimator attribute) (tutorials.examples.train_hypergrid_gafn.RND attribute) pretrain_prior() (in module tutorials.examples.train_diffusion_rtb) primes (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) print_final_results() (in module tutorials.examples.train_hypergrid_exploration_examples) print_hairpin_fe() (MFE_calculator.RNAFolderBase method) print_stack_fe() (MFE_calculator.RNAFolderBase method) print_term_fe() (MFE_calculator.RNAFolderBase method) prior (gfn.gym.diffusion_sampling.Posterior9of25GaussianMixture attribute) (gfn.gym.helpers.bayesian_structure.scores.BaseScore attribute) prior_pf (gfn.gflownet.RelativeTBBase property) (gfn.gflownet.trajectory_balance.RelativeTBBase property) prioritized_capacity (gfn.containers.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) prioritized_sampling (gfn.containers.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) process (gfn.gym.helpers.chip_design.plc_client.PlacementCost attribute) PROJECT_ROOT (in module tutorials.examples.train_with_compile) projection (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) protein_seq (env.CodonDesignEnv attribute) push_source_flow_to_terminal_states() (in module gfn.gym.helpers.bayesian_structure.jsd)

Q

q_proj (gfn.utils.modules._AutoregressiveTransformerBlock attribute) quarter_circ (gfn.gym.helpers.box_polar_utils.DistributionWrapper attribute), [1] (gfn.gym.helpers.box_utils.DistributionWrapper attribute), [1] quarter_disk (gfn.gym.helpers.box_polar_utils.DistributionWrapper attribute), [1] (gfn.gym.helpers.box_utils.DistributionWrapper attribute), [1]

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

R

R (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) R0 (gfn.gym.box.BoxPolar attribute), [1] (gfn.gym.BoxPolar attribute), [1] (gfn.gym.hypergrid.BitwiseXORReward attribute) (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) (gfn.gym.hypergrid.UniformRandomReward attribute) R1 (gfn.gym.box.BoxPolar attribute), [1] (gfn.gym.BoxPolar attribute), [1] R2 (gfn.gym.box.BoxPolar attribute), [1] (gfn.gym.BoxPolar attribute), [1] R_mode (gfn.gym.hypergrid.UniformRandomReward attribute) random_target_net (tutorials.examples.train_hypergrid_gafn.RND attribute) range (gfn.gym.chip_design.CostStats attribute) rank (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) read_node_order_file() (in module gfn.gym.helpers.chip_design.utils) reason (performance_tuning.BenchmarkResult attribute) recalc_logprobs (tutorials.examples.train_with_compile.TrainingComponents attribute) receivers_mlp (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) RecurrentDiscretePolicyEstimator (class in gfn.estimators) RecurrentDiscreteSequenceModel (class in gfn.utils.modules) RecurrentPolicyMixin (class in gfn.estimators) recv() (in module gfn.utils.distributed) reduction (gfn.estimators.ConditionalLogZEstimator attribute) (gfn.gflownet.mle.MLEDiffusion attribute) reduction_function (gfn.estimators.ConditionalScalarEstimator attribute), [1] (gfn.estimators.ScalarEstimator attribute), [1] REDUCTION_FUNCTIONS (in module gfn.estimators) RegressionLoss (class in gfn.gflownet) (class in gfn.gflownet.losses) RelativeLogPartitionVarianceGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) RelativeTBBase (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) RelativeTrajectoryBalanceGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) remote_buffer_freq (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) remote_manager_rank (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) (gfn.containers.ReplayBuffer attribute) render() (in module tutorials.examples.train_line) render_states() (in module tutorials.examples.train_graph_ring) (in module tutorials.examples.train_graph_triangle) ReplayBuffer (class in gfn.containers) (class in gfn.containers.replay_buffer) ReplayBufferManager (class in gfn.containers.replay_buffer_manager) report_load_imbalance() (in module gfn.utils.distributed) report_time_info() (in module gfn.utils.distributed) requires_logf (tutorials.examples.train_with_compile.FlowVariant attribute) reset() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.env.GraphEnv method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) reset_noise() (gfn.utils.modules.NoisyLinear method) reset_parameters() (gfn.utils.modules.NoisyLinear method) residual_dropout (gfn.utils.modules._AutoregressiveTransformerBlock attribute) resolve_device() (in module performance_tuning) (in module tutorials.examples.train_with_compile)

resolve_output_paths() (in module tutorials.examples.train_diffusion_rtb) restore_macro_orientations() (in module gfn.gym.helpers.chip_design.utils) restore_node_xy_coordinates() (in module gfn.gym.helpers.chip_design.utils) reverse_backward_trajectories() (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) reward() (env.CodonDesignEnv method) (gfn.env.Env method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.box.BoxPolar method) (gfn.gym.BoxPolar method) (gfn.gym.ConditionalHyperGrid method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.GraphBuilding method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.ConditionalHyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.NonAutoregressiveBitSequence method) (gfn.gym.perfect_tree.PerfectBinaryTree method) (gfn.gym.PerfectBinaryTree method) (gfn.gym.set_addition.SetAddition method) (gfn.gym.SetAddition method) reward_exponent (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute), [1] (gfn.gym.NonAutoregressiveBitSequence attribute), [1] reward_fn (gfn.gym.HyperGrid attribute), [1] (gfn.gym.hypergrid.HyperGrid attribute), [1] (gfn.gym.perfect_tree.PerfectBinaryTree attribute), [1] (gfn.gym.PerfectBinaryTree attribute), [1] (gfn.gym.set_addition.SetAddition attribute), [1] (gfn.gym.SetAddition attribute), [1] reward_fn_kwargs (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) reward_matching_loss() (gfn.gflownet.flow_matching.FMGFlowNet method) (gfn.gflownet.FMGFlowNet method) reward_norm (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) reward_scale (tutorials.examples.train_hypergrid_gafn.RND attribute) reward_temper (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) reward_val (tutorials.examples.train_graph_ring.RingReward attribute) (tutorials.examples.train_graph_triangle.TriangleReward attribute) RingReward (class in tutorials.examples.train_graph_ring) RNAFolder (class in MFE_calculator) RNAFolderBase (class in MFE_calculator) RNAFolderBase2 (class in MFE_calculator) RNASolution (class in MFE_calculator) RNASolutionBase (class in MFE_calculator) RNASolutionBase2 (class in MFE_calculator) RND (class in tutorials.examples.train_hypergrid_gafn) rnd (tutorials.examples.train_hypergrid_gafn.TBGAFN attribute) rnd_parameters() (tutorials.examples.train_hypergrid_gafn.TBGAFN method) rnn_type (gfn.utils.modules.RecurrentDiscreteSequenceModel attribute) RolloutContext (class in gfn.estimators) rtb_to_trust_pcl_params() (in module gfn.utils.trust_pcl) rule_targets (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) run() (gfn.containers.replay_buffer_manager.ReplayBufferManager method) run_benchmark() (in module performance_tuning) run_benchmarks() (in module performance_tuning) run_case() (in module tutorials.examples.train_with_compile)

S

s0 (gfn.env.DiscreteEnv attribute), [1] (gfn.env.Env attribute), [1] (gfn.env.GraphEnv attribute), [1] (gfn.gym.chip_design.ChipDesignStates attribute) (gfn.gym.perfect_tree.PerfectBinaryTree attribute) (gfn.gym.PerfectBinaryTree attribute) (gfn.states.GraphStates attribute), [1] (gfn.states.States attribute), [1] s_dim (gfn.estimators.DiffusionPolicyEstimator attribute) (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] s_emb_dim (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] s_model (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] (gfn.utils.modules.DiffusionPISStateEncoding attribute) sample() (gfn.containers.base.Container method) (gfn.containers.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) (gfn.gym.diffusion_sampling.BaseTarget method) (gfn.gym.diffusion_sampling.Funnel method) (gfn.gym.diffusion_sampling.Grid25GaussianMixture method) (gfn.gym.diffusion_sampling.ManyWell method) (gfn.gym.diffusion_sampling.Posterior9of25GaussianMixture method) (gfn.gym.diffusion_sampling.SimpleGaussianMixture method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution method) (gfn.gym.helpers.box_polar_utils.DistributionWrapper method) (gfn.gym.helpers.box_polar_utils.QuarterCircle method) (gfn.gym.helpers.box_polar_utils.QuarterCircleWithExit method) (gfn.gym.helpers.box_polar_utils.QuarterDisk method) (gfn.gym.helpers.box_utils.BoxCartesianDistribution method) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution 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.states.States method) (gfn.utils.distributions.GraphActionDistribution method) (gfn.utils.distributions.IsotropicGaussian method) (gfn.utils.distributions.UnsqueezedCategorical method) (tutorials.examples.train_line.ScaledGaussianWithOptionalExit method) sample_actions() (gfn.samplers.Sampler method) sample_conditions() (gfn.env.Env method) (gfn.gym.ConditionalHyperGrid method) (gfn.gym.hypergrid.ConditionalHyperGrid method) sample_erdos_renyi_graph() (in module gfn.gym.helpers.bayesian_structure.graph) sample_erdos_renyi_linear_gaussian() (in module gfn.gym.helpers.bayesian_structure.graph) sample_from_linear_gaussian() (in module gfn.gym.helpers.bayesian_structure.sampling) sample_from_reward() (in module tutorials.examples.train_box) (in module tutorials.examples.train_box_legacy) SAMPLE_INIT_PLACEMENT (in module gfn.gym.helpers.chip_design) SAMPLE_NETLIST_FILE (in module gfn.gym.helpers.chip_design) 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.mle.MLEDiffusion method) (gfn.gflownet.PFBasedGFlowNet method) (gfn.samplers.LocalSearchSampler method) (gfn.samplers.Sampler method) (in module tutorials.examples.train_with_compile) Sampler (class in gfn.samplers) sampler (gfn.gflownet.mle.MLEDiffusion attribute) (tutorials.examples.train_with_compile.TrainingComponents attribute) sampler_kwargs (tutorials.examples.train_with_compile.TrainingComponents attribute) save() (gfn.containers.base.Container method) (gfn.containers.Container method) (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method) save_node_order_file() (in module gfn.gym.helpers.chip_design.utils) save_placement() (in module gfn.gym.helpers.chip_design.utils) ScalarEstimator (class in gfn.estimators) scale (gfn.utils.distributions.IsotropicGaussian attribute), [1] ScaledGaussianWithOptionalExit (class in tutorials.examples.train_line) scales (gfn.gym.diffusion_sampling.ManyWell attribute) scoring_function (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) scoring_only (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) seed (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.gym.hypergrid.CorruptedReward attribute) (gfn.gym.hypergrid.UniformRandomReward attribute) send() (in module gfn.utils.distributed) send_termination_signal() (gfn.containers.replay_buffer_manager.ReplayBufferManager static method) senders_mlp (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) seq_length (env.CodonDesignEnv attribute) (preprocessor.CodonSequencePreprocessor attribute) seq_size (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute), [1] (gfn.gym.BitSequencePlus attribute) (gfn.gym.NonAutoregressiveBitSequence attribute), [1] serialize() (gfn.containers.message.Message method) set_exit_masks() (gfn.states.DiscreteStates method) set_nonexit_action_masks() (gfn.states.DiscreteStates method) set_seed() (in module gfn.utils.common) set_up_fm_gflownet() (in module tutorials.examples.train_hypergrid) set_up_gflownet() (in module tutorials.examples.train_hypergrid) set_up_logF_estimator() (in module tutorials.examples.train_hypergrid) set_up_pb_pf_estimators() (in module tutorials.examples.train_hypergrid) set_weights() (env.CodonDesignEnv method) SetAddition (class in gfn.gym) (class in gfn.gym.set_addition) sf (gfn.env.DiscreteEnv attribute), [1] (gfn.env.Env attribute), [1] (gfn.env.GraphEnv attribute), [1] (gfn.gym.chip_design.ChipDesignStates attribute) (gfn.gym.perfect_tree.PerfectBinaryTree attribute) (gfn.gym.PerfectBinaryTree attribute) (gfn.states.GraphStates attribute), [1] (gfn.states.States attribute), [1] shift_coeffs_per_rule (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) ShiftedCoshLoss (class in gfn.gflownet) (class in gfn.gflownet.losses) sigma (gfn.estimators.PinnedBrownianMotionBackward attribute) (gfn.estimators.PinnedBrownianMotionForward attribute) (gfn.gflownet.mle.MLEDiffusion attribute)

sigmas (gfn.gym.Line attribute), [1] (gfn.gym.line.Line attribute), [1] SIMPLE_PAIRING_FE (in module MFE_utils) SimpleGaussianMixture (class in gfn.gym.diffusion_sampling) SINGULARITY_DEF (in module gfn.gym.helpers.chip_design) sinusoidal_position_encoding() (in module gfn.utils.modules) SinusoidalPositionalEmbedding (class in gfn.utils.modules) skip_normalization (gfn.utils.modules.UniformModule attribute), [1] skipped (performance_tuning.BenchmarkResult attribute) sliced_log_reward() (in module gfn.gym.helpers.diffusion_utils) smallest_multiplier_to_integers() (in module gfn.gym.hypergrid) sock (gfn.gym.helpers.chip_design.plc_client.PlacementCost attribute) solve() (MFE_calculator.RNAFolder method) SparseReward (class in gfn.gym.hypergrid) split_PF_module_output() (in module gfn.gym.helpers.box_polar_utils) (in module gfn.gym.helpers.box_utils) SquaredLoss (class in gfn.gflownet) (class in gfn.gflownet.losses) stack() (gfn.actions.Actions class method) (gfn.gym.bitSequence.BitSequenceStates class method) (gfn.gym.chip_design.ChipDesignStates class method) (gfn.states.DiscreteStates class method) (gfn.states.GraphStates class method) (gfn.states.States class method) (gfn.utils.graphs.GeometricBatch class method) state_evaluator (gfn.gym.graph_building.GraphBuilding attribute), [1] (gfn.gym.GraphBuilding attribute), [1] state_evaluator() (gfn.gym.helpers.bayesian_structure.scores.BaseScore method) (gfn.gym.helpers.bayesian_structure.scores.BGeScore method) state_shape (gfn.env.DiscreteEnv attribute) (gfn.env.Env attribute), [1] (gfn.gym.chip_design.ChipDesignStates attribute) (gfn.states.States attribute), [1] states (bench_get_scores_all._DBDummyTransitions attribute) (bench_get_scores_all._ModDBDummyTransitions attribute) States (class in gfn.states) (env.CodonDesignEnv attribute) states (gfn.containers.states_container.StatesContainer attribute), [1] (gfn.containers.StatesContainer attribute), [1] (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] (gfn.containers.Transitions attribute), [1] (gfn.containers.transitions.Transitions attribute), [1] States (gfn.env.DiscreteEnv attribute) (gfn.env.Env attribute), [1] (gfn.env.GraphEnv attribute), [1] (gfn.gym.BitSequence attribute) (gfn.gym.bitSequence.BitSequence attribute) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute) (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) (gfn.gym.discrete_ebm.DiscreteEBM attribute) (gfn.gym.DiscreteEBM attribute) states (gfn.gym.helpers.box_cartesian_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianDistribution attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBDistribution attribute) States (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) (gfn.gym.NonAutoregressiveBitSequence attribute) (gfn.gym.perfect_tree.PerfectBinaryTree attribute) (gfn.gym.PerfectBinaryTree attribute) (gfn.gym.set_addition.SetAddition attribute) (gfn.gym.SetAddition attribute) states_actions_tns_to_traj() (in module gfn.utils.training) states_from_batch_shape() (gfn.env.DiscreteEnv method) (gfn.env.Env method) states_from_tensor() (gfn.env.DiscreteEnv method) (gfn.env.Env method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) StatesContainer (class in gfn.containers) (class in gfn.containers.states_container) StateType (in module gfn.containers.states_container) std (gfn.gym.chip_design.CostStats attribute) std_cell_placer_mode (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) std_init (gfn.utils.modules.NoisyLinear attribute) std_iter_ms (performance_tuning.BenchmarkResult attribute) step() (env.CodonDesignEnv method) (gfn.env.Env method) (gfn.env.GraphEnv method) (gfn.gym.bayesian_structure.BayesianStructure method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequence.BitSequencePlus method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.BitSequencePlus method) (gfn.gym.box.BoxPolar method) (gfn.gym.BoxPolar method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) (gfn.gym.diffusion_sampling.DiffusionSampling method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.graph_building.GraphBuilding method) (gfn.gym.GraphBuilding method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.Line method) (gfn.gym.line.Line method) (gfn.gym.NonAutoregressiveBitSequence method) (gfn.gym.perfect_tree.PerfectBinaryTree method) (gfn.gym.PerfectBinaryTree method) (gfn.gym.set_addition.SetAddition method) (gfn.gym.SetAddition method) StepEstimator (class in tutorials.examples.train_line) STOP_CODONS (in module utils) stop_mlp (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) store_all_states (gfn.gym.HyperGrid attribute), [1] (gfn.gym.hypergrid.HyperGrid attribute), [1] store_locally (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) str_as_pair_int() (in module MFE_utils) SubTBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.sub_trajectory_balance) summarize_results() (in module tutorials.examples.train_with_compile) supported_flows (tutorials.examples.train_with_compile.EnvironmentBenchmark attribute) syn_indices (env.CodonDesignEnv attribute) synchronize() (in module performance_tuning) synchronize_if_needed() (in module tutorials.examples.train_with_compile)

T

t (gfn.gym.helpers.bayesian_structure.scores.BGeScore attribute) t_emb_dim (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] t_model (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1] (gfn.utils.modules.DiffusionPISTimeEncoding attribute) t_scale (gfn.gflownet.mle.MLEDiffusion attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute) table (gfn.utils.modules.Tabular attribute), [1] Tabular (class in gfn.utils.modules) target (gfn.gym.diffusion_sampling.DiffusionSampling attribute), [1] target_dna (CAI_calculator.CAICalculator attribute) target_dtype (gfn.preprocessors.Preprocessor attribute) target_lcms (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) target_rna (CAI_calculator.CAICalculator attribute) TargetEntry (in module gfn.gym.diffusion_sampling) targets (gfn.gym.hypergrid.SparseReward attribute) TargetsTensor (in module gfn.gflownet.sub_trajectory_balance) TBGAFN (class in tutorials.examples.train_hypergrid_gafn) TBGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) temperature (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.BitSequencePlus attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator attribute) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator attribute) (tutorials.examples.train_bayesian_structure.DAGEdgeActionGNNv2 attribute) temporarily_set_seed() (in module gfn.utils.common) tensor (bench_get_scores_all._DBDummyActions attribute) (bench_get_scores_all._DBDummyStates attribute) (bench_get_scores_all._ModDBDummyActions attribute) (bench_get_scores_all._ModDBDummyStates attribute) (gfn.actions.Actions attribute), [1] (gfn.actions.GraphActions attribute), [1] (gfn.gym.bitSequence.BitSequenceStates attribute) (gfn.states.GraphStates property) (gfn.states.States attribute), [1] (gfn.utils.graphs.GeometricBatch attribute) term_states (gfn.gym.perfect_tree.PerfectBinaryTree attribute) (gfn.gym.PerfectBinaryTree attribute) terminating_idx (bench_get_scores_all._SubTBDummyTrajectories attribute) (gfn.containers.Trajectories attribute), [1] (gfn.containers.trajectories.Trajectories attribute), [1] terminating_log_rewards (gfn.containers.states_container.StatesContainer property) (gfn.containers.StatesContainer property) terminating_states (gfn.containers.base.Container property) (gfn.containers.Container property) (gfn.containers.replay_buffer.Container property) (gfn.containers.states_container.StatesContainer property) (gfn.containers.StatesContainer property) (gfn.containers.Trajectories property) (gfn.containers.trajectories.Trajectories property) (gfn.containers.Transitions property) (gfn.containers.transitions.Transitions property) (gfn.env.DiscreteEnv property) (gfn.gym.BitSequence property) (gfn.gym.bitSequence.BitSequence property) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence property) (gfn.gym.discrete_ebm.DiscreteEBM property) (gfn.gym.DiscreteEBM property) (gfn.gym.HyperGrid property) (gfn.gym.hypergrid.HyperGrid property) (gfn.gym.NonAutoregressiveBitSequence property) (gfn.gym.perfect_tree.PerfectBinaryTree property) (gfn.gym.PerfectBinaryTree property) (gfn.gym.set_addition.SetAddition property) (gfn.gym.SetAddition property) TerminatingStateBuffer (class in gfn.containers) (class in gfn.containers.replay_buffer) test_backward_step_subtracts_action() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianEnvironment method) test_backward_trajectory_reaches_s0() (test_box_cartesian_utils.TestBoxCartesianEndToEnd method) (test_box_polar_utils.TestBoxCartesianEndToEnd method) test_batch_mixed_valid_invalid() (test_box_cartesian_utils.TestBoxCartesianDistribution method) test_box_cartesian_utils module test_box_polar_utils module test_exit_action_valid() (test_box_polar_utils.TestBoxCartesianEnvironment method) test_exit_action_valid_from_non_s0() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) test_exit_from_s0_invalid() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) test_exit_from_s0_invalid_all_exits_batch() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) test_exit_from_s0_invalid_mixed_batch() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) test_exit_from_s0_neg_inf() (test_box_cartesian_utils.TestBoxCartesianDistribution method) test_exit_valid_mixed_batch_no_s0() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) test_forward_backward_roundtrip() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianEnvironment method) test_forward_step_adds_action() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianEnvironment method) test_impossible_action_gets_neg_inf() (test_box_cartesian_utils.TestBoxCartesianDistribution method) test_is_action_valid_boundary() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianEnvironment method) test_is_action_valid_non_s0_forward() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianEnvironment method) test_is_action_valid_s0_forward() (test_box_cartesian_utils.TestBoxCartesianEnvironment method) (test_box_polar_utils.TestBoxCartesianEnvironment method) test_log_prob_exit_at_boundary() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) test_log_prob_finite() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) test_log_prob_near_origin() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) test_mixed_distributions() (in module test_box_polar_utils) test_no_exit_from_s0() (test_box_polar_utils.TestBoxCartesianDistribution method) test_non_exit_at_boundary_neg_inf() (test_box_cartesian_utils.TestBoxCartesianDistribution method) test_out_of_range_backward_action_gets_neg_inf() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) test_reward_function() (test_box_cartesian_utils.TestBoxCartesianEndToEnd method) (test_box_polar_utils.TestBoxCartesianEndToEnd method) test_sample_mixed_dims_near_origin() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) test_sample_near_origin_to_s0() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) test_sample_non_origin_valid() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) test_sample_non_s0_in_range() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) test_sample_s0_in_range() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) test_sample_shape() (test_box_cartesian_utils.TestBoxCartesianDistribution method) (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) (test_box_polar_utils.TestBoxCartesianDistribution method) (test_box_polar_utils.TestBoxCartesianPBDistribution method) test_trajectory_stays_in_bounds() (test_box_cartesian_utils.TestBoxCartesianEndToEnd method) (test_box_polar_utils.TestBoxCartesianEndToEnd method) test_valid_action_gets_finite_log_prob() (test_box_cartesian_utils.TestBoxCartesianDistribution method) test_valid_backward_action_gets_finite_log_prob() (test_box_cartesian_utils.TestBoxCartesianPBDistribution method) TestBoxCartesianDistribution (class in test_box_cartesian_utils) (class in test_box_polar_utils) TestBoxCartesianEndToEnd (class in test_box_cartesian_utils) (class in test_box_polar_utils) TestBoxCartesianEnvironment (class in test_box_cartesian_utils) (class in test_box_polar_utils) TestBoxCartesianPBDistribution (class in test_box_cartesian_utils) (class in test_box_polar_utils) threshold_metrics() (in module gfn.gym.helpers.bayesian_structure.evaluation) tier_indicators() (gfn.gym.hypergrid.BitwiseXORReward method) (gfn.gym.hypergrid.ConditionalMultiScaleReward method) (gfn.gym.hypergrid.MultiplicativeCoprimeReward method) tier_weights (gfn.gym.hypergrid.BitwiseXORReward attribute) (gfn.gym.hypergrid.ConditionalMultiScaleReward attribute) (gfn.gym.hypergrid.MultiplicativeCoprimeReward attribute) time_iterations() (in module performance_tuning) Timer (class in gfn.utils.common) timestep_phase (gfn.utils.modules.DiffusionPISTimeEncoding attribute) timing (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.replay_buffer_manager.ReplayBufferManager attribute) (gfn.containers.ReplayBuffer attribute) timing_data (gfn.containers.replay_buffer.ReplayBuffer attribute) (gfn.containers.ReplayBuffer attribute) timing_dict (gfn.utils.common.Timer attribute) timing_log() (gfn.containers.replay_buffer.ReplayBuffer method) (gfn.containers.ReplayBuffer method)

timing_print_summary() (in module gfn.utils.common) to() (gfn.states.DiscreteStates method) (gfn.states.GraphStates method) (gfn.states.States method) to_dict() (gfn.gym.helpers.bayesian_structure.jsd.FullPosterior method) (gfn.gym.helpers.bayesian_structure.jsd.GraphCollection method) to_mRNA_string() (in module utils) to_probability_distribution() (bench_get_scores_all._ModDBDummyEstimator method) (gfn.estimators.ConditionalScalarEstimator method) (gfn.estimators.DiffusionPolicyEstimator method) (gfn.estimators.DiscreteGraphPolicyEstimator method) (gfn.estimators.DiscretePolicyEstimator method) (gfn.estimators.Estimator method) (gfn.estimators.PinnedBrownianMotionBackward method) (gfn.estimators.PinnedBrownianMotionForward method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPFEstimator method) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator method) (gfn.gym.helpers.box_polar_utils.BoxPFEstimator method) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator method) (gfn.gym.helpers.box_utils.BoxCartesianPFEstimator method) (gfn.gym.helpers.box_utils.BoxPBEstimator method) (gfn.gym.helpers.box_utils.BoxPFEstimator method) (tutorials.examples.train_line.StepEstimator method) to_states_container() (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) to_str() (gfn.gym.bitSequence.BitSequenceStates method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequenceStates method) to_tensordict() (gfn.containers.base.Container method) (gfn.containers.Container method) (gfn.containers.states_container.StatesContainer method) (gfn.containers.StatesContainer method) (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) (gfn.containers.Transitions method) (gfn.containers.transitions.Transitions 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.mle.MLEDiffusion method) (gfn.gflownet.ModifiedDBGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) to_transitions() (gfn.containers.Trajectories method) (gfn.containers.trajectories.Trajectories method) token_embedding (gfn.utils.modules.TransformerDiscreteSequenceModel attribute) tokenize_sequence_to_tensor() (in module utils) topological_sort_data() (in module gfn.gym.helpers.bayesian_structure.sampling) train() (in module tutorials.examples.train_conditional) (in module tutorials.examples.train_hypergrid_exploration_examples) (in module tutorials.examples.train_line) train_global_group (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute) training_container (gfn.containers.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.NormBasedDiversePrioritizedReplayBuffer attribute) (gfn.containers.replay_buffer.ReplayBuffer attribute), [1] (gfn.containers.replay_buffer.TerminatingStateBuffer attribute), [1] (gfn.containers.ReplayBuffer attribute), [1] (gfn.containers.TerminatingStateBuffer attribute), [1] training_loop() (in module tutorials.examples.train_with_compile) TrainingComponents (class in tutorials.examples.train_with_compile) TrainingSampleType (in module gfn.gflownet.base) Trajectories (class in gfn.containers) (class in gfn.containers.trajectories) trajectory_estimator_outputs (gfn.estimators.RolloutContext attribute) trajectory_from_terminating_states() (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequence.BitSequencePlus method) (gfn.gym.BitSequencePlus method) trajectory_log_probs (gfn.estimators.RolloutContext attribute) trajectory_log_probs_backward() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) trajectory_log_probs_forward() (gfn.gflownet.base.TrajectoryBasedGFlowNet method) (gfn.gflownet.TrajectoryBasedGFlowNet method) TrajectoryBasedGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.base) TransformerDiscreteSequenceModel (class in gfn.utils.modules) transition_table (gfn.gym.perfect_tree.PerfectBinaryTree attribute) (gfn.gym.PerfectBinaryTree attribute) Transitions (class in gfn.containers) (class in gfn.containers.transitions) TriangleReward (class in tutorials.examples.train_graph_triangle) true_dist() (gfn.env.Env method) (gfn.gym.BitSequence method) (gfn.gym.bitSequence.BitSequence method) (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence method) (gfn.gym.ConditionalHyperGrid method) (gfn.gym.discrete_ebm.DiscreteEBM method) (gfn.gym.DiscreteEBM method) (gfn.gym.HyperGrid method) (gfn.gym.hypergrid.ConditionalHyperGrid method) (gfn.gym.hypergrid.HyperGrid method) (gfn.gym.NonAutoregressiveBitSequence method) trust_pcl_to_rtb_params() (in module gfn.utils.trust_pcl) TrustPCLGFlowNet (class in gfn.gflownet) (class in gfn.gflownet.trajectory_balance) try_compile_gflownet() (in module gfn.utils.compile) tuple_as_pair_int() (in module MFE_utils) TURNER_FE (in module MFE_utils) tutorials module tutorials.examples module tutorials.examples.train_bayesian_structure module tutorials.examples.train_bit_sequences module tutorials.examples.train_bitsequence_non_autoregressive module tutorials.examples.train_bitsequence_recurrent module tutorials.examples.train_box module tutorials.examples.train_box_legacy module tutorials.examples.train_chip_design module tutorials.examples.train_chip_design_medium module tutorials.examples.train_conditional module tutorials.examples.train_diffusion_rtb module tutorials.examples.train_diffusion_sampler module tutorials.examples.train_discreteebm module tutorials.examples.train_graph_ring module tutorials.examples.train_graph_triangle module tutorials.examples.train_hypergrid module tutorials.examples.train_hypergrid_buffer module tutorials.examples.train_hypergrid_exploration_examples module tutorials.examples.train_hypergrid_gafn module tutorials.examples.train_hypergrid_local_search module tutorials.examples.train_hypergrid_simple module tutorials.examples.train_ising module tutorials.examples.train_line module tutorials.examples.train_with_compile module tutorials.examples.train_with_example_modes module

U

U (MFE_calculator.RNASolutionBase attribute) undirected_reward() (tutorials.examples.train_graph_ring.RingReward method) uniform() (gfn.estimators.DiscretePolicyEstimator class method) (gfn.gym.helpers.box_cartesian_utils.BoxCartesianPBEstimator class method) (gfn.gym.helpers.box_polar_utils.BoxPBEstimator class method) (gfn.gym.helpers.box_utils.BoxCartesianPBEstimator class method) (gfn.gym.helpers.box_utils.BoxPBEstimator class method) UniformBoxCartesianPBModule (class in gfn.gym.helpers.box_cartesian_utils) (class in gfn.gym.helpers.box_utils) UniformModule (class in gfn.utils.modules)

UniformPrior (class in gfn.gym.helpers.bayesian_structure.priors) UniformRandomReward (class in gfn.gym.hypergrid) UnsqueezedCategorical (class in gfn.utils.distributions) update_masks() (env.CodonDesignEnv method) (gfn.gym.chip_design.ChipDesign method) (gfn.gym.ChipDesign method) use_compile (performance_tuning.BenchmarkConfig attribute) use_edge_ri (tutorials.examples.train_hypergrid_gafn.TBGAFN attribute) use_training_samples (tutorials.examples.train_with_compile.TrainingComponents attribute) utils module

V

V (MFE_calculator.RNASolutionBase attribute) v_proj (gfn.utils.modules._AutoregressiveTransformerBlock attribute) v_soft_s0 (gfn.gflownet.trajectory_balance.TrustPCLGFlowNet property) (gfn.gflownet.TrustPCLGFlowNet property) validate() (gfn.env.DiscreteEnv method) (in module gfn.utils.training) validate_modes (gfn.gym.HyperGrid attribute) (gfn.gym.hypergrid.HyperGrid attribute) validate_policy_estimator() (in module gfn.estimators) value_names (gfn.utils.modules.TransformerDiscreteSequenceModel attribute)

VARIANT_COLORS (in module tutorials.examples.train_with_compile) visualize() (gfn.gym.diffusion_sampling.BaseTarget method) (gfn.gym.diffusion_sampling.Funnel method) (gfn.gym.diffusion_sampling.Grid25GaussianMixture method) (gfn.gym.diffusion_sampling.ManyWell method) (gfn.gym.diffusion_sampling.Posterior9of25GaussianMixture method) (gfn.gym.diffusion_sampling.SimpleGaussianMixture method) viz_2d_slice() (in module gfn.gym.helpers.diffusion_utils) vocab_size (gfn.utils.modules.AutoregressiveDiscreteSequenceModel property) (gfn.utils.modules.RecurrentDiscreteSequenceModel property) (gfn.utils.modules.TransformerDiscreteSequenceModel property)

W

wait() (gfn.utils.distributed.AsyncSendHandle method) walk_range() (MFE_calculator.RNASolution method) walk_range_ex() (MFE_calculator.RNASolution method) warm_up() (in module gfn.utils.training) warmup_iters (performance_tuning.BenchmarkSettings attribute) warn_about_recalculating_logprobs() (in module gfn.utils.handlers) weight (gfn.utils.modules.NoisyLinear property) weight_mu (gfn.utils.modules.NoisyLinear attribute) weight_sigma (gfn.utils.modules.NoisyLinear attribute) weighting (gfn.gflownet.sub_trajectory_balance.SubTBGFlowNet attribute), [1] (gfn.gflownet.SubTBGFlowNet attribute), [1] weights (env.CodonDesignEnv attribute) wirelength_weight (gfn.gym.chip_design.ChipDesign attribute) (gfn.gym.ChipDesign attribute) word_size (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.bitSequence.BitSequenceStates attribute), [1] (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute), [1] (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequenceStates attribute), [1] (gfn.gym.BitSequencePlus attribute) (gfn.gym.NonAutoregressiveBitSequence attribute), [1]

words_per_seq (gfn.gym.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequence attribute), [1] (gfn.gym.bitSequence.BitSequencePlus attribute) (gfn.gym.bitSequence.BitSequenceStates attribute), [1] (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequence attribute), [1] (gfn.gym.bitSequenceNonAutoregressive.NonAutoregressiveBitSequenceStates attribute), [1] (gfn.gym.BitSequencePlus attribute) (gfn.gym.NonAutoregressiveBitSequence attribute), [1] world_size (gfn.utils.distributed.DistributedContext attribute) (gfn.utils.distributed.DistributedContextMPI4Py attribute)

Z

Z_x1 (gfn.gym.diffusion_sampling.ManyWell attribute) Z_x2 (gfn.gym.diffusion_sampling.ManyWell attribute)

zero_init (gfn.utils.modules.DiffusionPISGradNetBackward attribute) (gfn.utils.modules.DiffusionPISGradNetForward attribute), [1]