Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | X | Z A ACT_MAP (in module dymad.modules.helpers) adj_to_edge() (in module dymad.utils) (in module dymad.utils.graph) AE_MAP (in module dymad.modules.collections) aggregate_cv_results() (in module dymad.training) (in module dymad.training.helper) all_results (dymad.training.driver.CVSearchRunResult attribute) allowed_predictors (dymad.models.model_spec.RolloutSpec attribute) (dymad.models.RolloutSpec attribute) alpha (dymad.losses.losses.WMSELoss attribute) (dymad.losses.WMSELoss attribute) AnalysisPhaseSpec (class in dymad.training) (class in dymad.training.phases) animate() (in module dymad.utils) (in module dymad.utils.plot) apply_data_transformations() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.trajectory_manager.TrajectoryManagerGraph method) (dymad.io.TrajectoryManager method) (dymad.io.TrajectoryManagerGraph method) apply_obs() (dymad.io.checkpoint.DataInterface method) (dymad.io.DataInterface method) (dymad.sako.adapter.SpectralRuntime method) apply_to_config() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) artifact_keys (dymad.training.phase_runtime.PhaseRecord attribute) (dymad.training.PhaseRecord attribute) ArtifactRegistry (class in dymad.training) (class in dymad.training.phase_runtime) artifacts (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseResult attribute) B backward() (dymad.utils.wrapper.JaxMultiInFn static method) base_dynamics() (dymad.models.recipes_corr.TemplateCorrAlg method) (dymad.models.recipes_corr.TemplateCorrDif method) BaseOptimizerPhase (class in dymad.training.phases) BasePhase (class in dymad.training.phases) BasePhaseSpec (class in dymad.training.phases) batch_to_runtime() (in module dymad.training.batch_adapter) best_loss (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.phase_runtime.TrainingHistoryArtifact attribute) (dymad.training.TrainerState attribute) (dymad.training.TrainingHistoryArtifact attribute) best_model_state_dict (dymad.training.phase_runtime.TrainingHistoryArtifact attribute) (dymad.training.TrainingHistoryArtifact attribute) BestModelExportPhase (class in dymad.training.phases) BoundaryLoadTrace (class in dymad.io) (class in dymad.io.checkpoint) bounded_nelder_mead_search_points() (in module dymad.training.helper) build() (dymad.models.runtime_view.ComponentInputView class method) build_autoencoder() (in module dymad.models.helpers) build_component_input_view() (in module dymad.models.runtime_view) build_initial_trainer_state() (in module dymad.training.phase_runtime) build_model() (in module dymad.models) (in module dymad.models.helpers) build_model_from_spec() (in module dymad.models) (in module dymad.models.helpers) build_phase() (dymad.training.phase_pipeline.PhasePipeline method) (dymad.training.PhasePipeline method) (in module dymad.training.phases) build_processor() (in module dymad.models.helpers) build_spectral_snapshot() (in module dymad.sako.snapshot) C CD_KM (class in dymad.models.recipes) CD_KMM (class in dymad.models.recipes) CD_KMSK (class in dymad.models.recipes) CD_LDM (class in dymad.models.recipes) CD_LFM (class in dymad.models.recipes) CD_SDM (class in dymad.models.recipes) central_diff() (in module dymad.numerics) (in module dymad.numerics.gradients) check_direction() (in module dymad.numerics) (in module dymad.numerics.linalg) check_linear_impl() (in module dymad.training.ls_update) check_linear_solve() (in module dymad.training.ls_update) check_orthogonality() (in module dymad.numerics) (in module dymad.numerics.linalg) checkpoint_file() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) checkpoint_path (dymad.sako.snapshot.SpectralSnapshot attribute) checkpoint_paths (dymad.training.CVResult attribute) (dymad.training.helper.CVResult attribute) checkpoint_payload() (dymad.training.ArtifactRegistry method) (dymad.training.phase_runtime.ArtifactRegistry method) (dymad.training.phase_runtime.TrainerState method) (dymad.training.TrainerState method) checkpoint_prefix (dymad.training.execution_services.ExecutionServices attribute) (dymad.training.ExecutionServices attribute) chirp_control() (in module dymad.utils.sampling) clip_eig() (in module dymad.sako.sako) compare_contour() (in module dymad.utils) (in module dymad.utils.plot) compDist() (in module dymad.numerics.dm) completed_epoch (dymad.training.phase_runtime.PhaseRecord attribute) (dymad.training.PhaseRecord attribute) complex_grid() (in module dymad.numerics) (in module dymad.numerics.complex) complex_plot() (in module dymad.numerics) (in module dymad.numerics.complex) complex_step() (in module dymad.numerics) (in module dymad.numerics.gradients) ComponentInputView (class in dymad.models.runtime_view) ComposedDynamics (class in dymad.models.model_base) Composer (in module dymad.models.model_base) compute_newton_cotes_weights() (in module dymad.numerics.weak) config (dymad.training.ModelArtifact attribute) (dymad.training.phase_runtime.ModelArtifact attribute) (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.phases.AnalysisPhaseSpec attribute) (dymad.training.phases.BasePhaseSpec attribute) (dymad.training.phases.DataPhaseSpec attribute) (dymad.training.phases.ExportPhaseSpec attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) (dymad.training.phases.OptimizerPhaseSpec attribute) (dymad.training.TrainerState attribute) config_logger() (in module dymad.utils) (in module dymad.utils.misc) configure_logger() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) CONT (dymad.models.model_base.ComposedDynamics attribute) (dymad.models.recipes.CD_KMM attribute) (dymad.models.recipes_corr.TemplateCorrAlg attribute) (dymad.models.recipes_corr.TemplateCorrDif attribute) ContextDataPhase (class in dymad.training.phases) continuous_time (dymad.models.model_spec.ModelSpec property) (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.ModelSpec property) (dymad.models.ResolvedModelSpec attribute) control (dymad.models.runtime_view.ComponentInputView property) ControlInterpolator (class in dymad.utils) (class in dymad.utils.control) converged (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.TrainerState attribute) convergence_epoch (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.phase_runtime.TrainingHistoryArtifact attribute) (dymad.training.TrainerState attribute) (dymad.training.TrainingHistoryArtifact attribute) create_dataloaders() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.trajectory_manager.TrajectoryManagerGraph method) (dymad.io.TrajectoryManager method) (dymad.io.TrajectoryManagerGraph method) create_graph_series_dataset() (dymad.io.trajectory_manager.TrajectoryManagerGraph method) (dymad.io.TrajectoryManagerGraph method) create_regular_series_dataset() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.TrajectoryManager method) crit (dymad.training.phase_runtime.TrainingHistoryArtifact attribute) (dymad.training.TrainingHistoryArtifact attribute) criteria (dymad.training.OptimizerStateArtifact attribute) (dymad.training.phase_runtime.OptimizerStateArtifact attribute) criteria_names (dymad.training.OptimizerStateArtifact attribute) (dymad.training.phase_runtime.OptimizerStateArtifact attribute) criteria_weights (dymad.training.OptimizerStateArtifact attribute) (dymad.training.phase_runtime.OptimizerStateArtifact attribute) criterion_name (dymad.training.EvaluationArtifact attribute) (dymad.training.phase_runtime.EvaluationArtifact attribute) CTRL_MAP (in module dymad.utils.sampling) CV_SEARCH_HANDLERS (dymad.training.driver.DriverBase attribute) (dymad.training.DriverBase attribute) CVResult (class in dymad.training) (class in dymad.training.helper) CVSearchBoundSpec (class in dymad.training.driver) CVSearchRunResult (class in dymad.training.driver) D data_truncation() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.trajectory_manager.TrajectoryManagerGraph method) (dymad.io.TrajectoryManager method) (dymad.io.TrajectoryManagerGraph method) DataInterface (class in dymad.io) (class in dymad.io.checkpoint) DataPhaseSpec (class in dymad.training) (class in dymad.training.phases) dec_auto() (in module dymad.models.components) dec_graph() (in module dymad.models.components) dec_iden() (in module dymad.models.components) DEC_MAP (in module dymad.models.components) dec_node() (in module dymad.models.components) decode() (dymad.io.checkpoint.DataInterface method) (dymad.io.DataInterface method) decoder (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) Decoder (in module dymad.models.model_base) decoder() (dymad.models.model_base.ComposedDynamics method) decoder_key (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) DecoderSpec (class in dymad.models) (class in dymad.models.model_spec) default_predictor (dymad.models.model_spec.RolloutSpec attribute) (dymad.models.RolloutSpec attribute) denoise() (in module dymad.numerics) (in module dymad.numerics.denoising) denoising_metrics() (in module dymad.numerics) (in module dymad.numerics.denoising) device (dymad.training.execution_services.ExecutionServices attribute) (dymad.training.ExecutionServices attribute) (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.TrainerState attribute) DGKBF (in module dymad.models.collections) DGKM (in module dymad.models.collections) DGKMSK (in module dymad.models.collections) DGLDM (in module dymad.models.collections) DGLTI (in module dymad.models.collections) diagnostic_info() (dymad.models.model_base.ComposedDynamics method) (dymad.models.recipes_corr.TemplateCorrDif method) (dymad.modules.GNN method) (dymad.modules.gnn.GNN method) (dymad.modules.misc.TakeFirst method) (dymad.modules.MLP method) (dymad.modules.mlp.MLP method) (dymad.utils.scheduler.SchedulerBase method) (dymad.utils.scheduler.StandardScheduler method) (dymad.utils.scheduler.SweepScheduler method) DimensionEstimator (class in dymad.numerics) (class in dymad.numerics.manifold) dims (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) disc2cont() (in module dymad.numerics) (in module dymad.numerics.complex) discrete_eigs (dymad.sako.adapter.SpectralEigensystem attribute) DKBF (in module dymad.models.collections) DKM (in module dymad.models.collections) DKMSK (in module dymad.models.collections) DLDM (in module dymad.models.collections) DLDMG (in module dymad.models.collections) DLTI (in module dymad.models.collections) DM (class in dymad.numerics) (class in dymad.numerics.dm) DMF (class in dymad.numerics) (class in dymad.numerics.dm) DriverBase (class in dymad.training) (class in dymad.training.driver) dS() (in module dymad.numerics.dm) DSDM (in module dymad.models.collections) DSDMG (in module dymad.models.collections) dt (dymad.sako.adapter.SpectralEigensystem attribute) dtype (dymad.training.ModelArtifact attribute) (dymad.training.phase_runtime.ModelArtifact attribute) dymad.io module dymad.io.checkpoint module dymad.io.series_adapter module dymad.io.trajectory_manager module dymad.losses module dymad.losses.losses module dymad.models module dymad.models.collections module dymad.models.components module dymad.models.helpers module dymad.models.model_base module dymad.models.model_spec module dymad.models.prediction module dymad.models.recipes module dymad.models.recipes_corr module dymad.models.rollout_engine module dymad.models.runtime_view module dymad.modules module dymad.modules.collections module dymad.modules.gnn module dymad.modules.helpers module dymad.modules.kernel module dymad.modules.krr module dymad.modules.linear module dymad.modules.misc module dymad.modules.mlp module dymad.modules.sequential module dymad.numerics module dymad.numerics.complex module dymad.numerics.denoising module dymad.numerics.dm module dymad.numerics.gradients module dymad.numerics.linalg module dymad.numerics.manifold module dymad.numerics.spectrum module dymad.numerics.time_int module dymad.numerics.weak module dymad.sako module dymad.sako.adapter module dymad.sako.base module dymad.sako.plotting module dymad.sako.rals module dymad.sako.sako module dymad.sako.snapshot module dymad.training module dymad.training.batch_adapter module dymad.training.driver module dymad.training.execution_services module dymad.training.helper module dymad.training.ls_update module dymad.training.phase_pipeline module dymad.training.phase_runtime module dymad.training.phases module dymad.training.trainer module dymad.training.trainer_run module dymad.utils module dymad.utils.control module dymad.utils.graph module dymad.utils.misc module dymad.utils.plot module dymad.utils.sampling module dymad.utils.scheduler module dymad.utils.wrapper module dyn_direct() (in module dymad.models.components) dyn_graph_direct() (in module dymad.models.components) dyn_graph_skip() (in module dymad.models.components) DYN_MAP (in module dymad.models.components) dyn_skip() (in module dymad.models.components) dynamics (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) dynamics() (dymad.models.model_base.ComposedDynamics method) (dymad.models.recipes.CD_SDM method) (dymad.models.recipes_corr.TemplateCorrAlg method) (dymad.models.recipes_corr.TemplateCorrDif method) dynamics_key (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) DynamicsSpec (class in dymad.models) (class in dymad.models.model_spec) E edge_attr (dymad.models.runtime_view.ComponentInputView property) edge_index (dymad.models.runtime_view.ComponentInputView property) edge_weight (dymad.models.runtime_view.ComponentInputView property) eig_low_rank() (in module dymad.numerics) (in module dymad.numerics.linalg) ell (dymad.modules.kernel.KernelScExp property) (dymad.modules.kernel.KernelScRBF property) (dymad.modules.KernelScExp property) (dymad.modules.KernelScRBF property) enc_corr_dif_auto() (in module dymad.models.recipes_corr) enc_corr_dif_ctrl() (in module dymad.models.recipes_corr) enc_graph_auto() (in module dymad.models.components) enc_graph_ctrl() (in module dymad.models.components) enc_graph_iden() (in module dymad.models.components) enc_iden() (in module dymad.models.components) ENC_MAP (in module dymad.models.components) enc_node_auto() (in module dymad.models.components) enc_node_ctrl() (in module dymad.models.components) enc_node_raw_ctrl() (in module dymad.models.components) enc_raw_ctrl() (in module dymad.models.components) enc_smpl_auto() (in module dymad.models.components) enc_smpl_ctrl() (in module dymad.models.components) encode() (dymad.io.checkpoint.DataInterface method) (dymad.io.DataInterface method) encode_runtime_batch() (in module dymad.sako.base) encoded_p0 (dymad.sako.snapshot.SpectralSnapshot attribute) encoded_p1 (dymad.sako.snapshot.SpectralSnapshot attribute) encoder (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) Encoder (in module dymad.models.model_base) encoder() (dymad.models.model_base.ComposedDynamics method) encoder_key (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) EncoderSpec (class in dymad.models) (class in dymad.models.model_spec) ensure_artifact_dirs() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) epoch (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.TrainerState attribute) epoch_times (dymad.training.phase_runtime.TrainingHistoryArtifact attribute) (dymad.training.TrainingHistoryArtifact attribute) eps (dymad.modules.kernel.KernelScDM property) (dymad.modules.KernelScDM property) epsOpt() (in module dymad.numerics.dm) estimate_measure() (dymad.sako.adapter.SpectralAnalysisAdapter method) (dymad.sako.base.SpectralAnalysis method) (dymad.sako.SAKO method) (dymad.sako.sako.SAKO method) estimate_ps() (dymad.sako.adapter.SpectralAnalysisAdapter method) (dymad.sako.base.SpectralAnalysis method) estimate_pseudospectrum() (in module dymad.sako.rals) estimate_residual() (dymad.sako.SAKO method) (dymad.sako.sako.SAKO method) eval_batch() (dymad.training.ls_update.LSUpdater method) eval_eigfun() (dymad.sako.base.SpectralAnalysis method) eval_eigfun_par() (dymad.sako.base.SpectralAnalysis method) eval_eigfunc_jac() (dymad.sako.adapter.SpectralAnalysisAdapter method) (dymad.sako.base.SpectralAnalysis method) eval_eigmode_jac() (dymad.sako.adapter.SpectralAnalysisAdapter method) (dymad.sako.base.SpectralAnalysis method) evaluate_all (dymad.training.AnalysisPhaseSpec attribute) (dymad.training.phases.AnalysisPhaseSpec attribute) EvaluationArtifact (class in dymad.training) (class in dymad.training.phase_runtime) execute() (dymad.training.phases.BaseOptimizerPhase method) (dymad.training.phases.BasePhase method) (dymad.training.phases.BestModelExportPhase method) (dymad.training.phases.ContextDataPhase method) (dymad.training.phases.LinearSolvePhase method) (dymad.training.phases.RunCheckpointExportPhase method) (dymad.training.phases.SummaryExportPhase method) (dymad.training.phases.ValidationAnalysisPhase method) execution_services (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.TrainerState attribute) ExecutionServices (class in dymad.training) (class in dymad.training.execution_services) expm_full_rank() (in module dymad.numerics) (in module dymad.numerics.linalg) expm_low_rank() (in module dymad.numerics) (in module dymad.numerics.linalg) export_kind (dymad.training.ExportPhaseSpec attribute) (dymad.training.phases.ExportPhaseSpec attribute) ExportArtifact (class in dymad.training) (class in dymad.training.phase_runtime) ExportPhaseSpec (class in dymad.training) (class in dymad.training.phases) extra_setup() (dymad.models.recipes_corr.TemplateCorrAlg method) (dymad.models.recipes_corr.TemplateCorrDif method) F family (dymad.models.DecoderSpec property) (dymad.models.DynamicsSpec property) (dymad.models.EncoderSpec property) (dymad.models.FeatureSpec property) (dymad.models.MemorySpec attribute) (dymad.models.model_spec.DecoderSpec property) (dymad.models.model_spec.DynamicsSpec property) (dymad.models.model_spec.EncoderSpec property) (dymad.models.model_spec.FeatureSpec property) (dymad.models.model_spec.MemorySpec attribute) (dymad.models.model_spec.RolloutSpec attribute) (dymad.models.RolloutSpec attribute) fe_step() (in module dymad.numerics) (in module dymad.numerics.time_int) feature (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) feature_key (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) Features (in module dymad.models.model_base) FeatureSpec (class in dymad.models) (class in dymad.models.model_spec) fenc_step() (dymad.models.recipes.CD_KMM method) filter_spectrum() (dymad.sako.base.SpectralAnalysis method) (in module dymad.sako) (in module dymad.sako.base) fit() (dymad.modules.krr.KRRBase method) (dymad.modules.KRRBase method) (dymad.numerics.DM method) (dymad.numerics.dm.DM method) (dymad.numerics.dm.DMF method) (dymad.numerics.dm.VBDM method) (dymad.numerics.DMF method) (dymad.numerics.VBDM method) fit_krr() (dymad.numerics.dm.DMF method) (dymad.numerics.DMF method) fit_transform() (dymad.numerics.DM method) (dymad.numerics.dm.DM method) (dymad.numerics.dm.DMF method) (dymad.numerics.dm.VBDM method) (dymad.numerics.DMF method) (dymad.numerics.VBDM method) flatten_nodes() (dymad.models.runtime_view.ComponentInputView method) FlexLinear (class in dymad.modules) (class in dymad.modules.linear) fold_metrics (dymad.training.CVResult attribute) (dymad.training.helper.CVResult attribute) forward() (dymad.losses.losses.VPTLoss method) (dymad.losses.losses.WMSELoss method) (dymad.losses.VPTLoss method) (dymad.losses.WMSELoss method) (dymad.models.model_base.ComposedDynamics method) (dymad.modules.FlexLinear method) (dymad.modules.GNN method) (dymad.modules.gnn.GNN method) (dymad.modules.gnn.IdenCatGNN method) (dymad.modules.gnn.ResBlockGNN method) (dymad.modules.IdenCatGNN method) (dymad.modules.IdenCatMLP method) (dymad.modules.kernel.KernelAbstract method) (dymad.modules.kernel.KernelOpSeparable method) (dymad.modules.kernel.KernelOpTangent method) (dymad.modules.kernel.KernelScDM method) (dymad.modules.kernel.KernelScExp method) (dymad.modules.kernel.KernelScRBF method) (dymad.modules.KernelAbstract method) (dymad.modules.KernelOpSeparable method) (dymad.modules.KernelOpTangent method) (dymad.modules.KernelScDM method) (dymad.modules.KernelScExp method) (dymad.modules.KernelScRBF method) (dymad.modules.krr.KRRBase method) (dymad.modules.KRRBase method) (dymad.modules.linear.FlexLinear method) (dymad.modules.misc.TakeFirst method) (dymad.modules.misc.TakeFirstGraph method) (dymad.modules.MLP method) (dymad.modules.mlp.IdenCatMLP method) (dymad.modules.mlp.MLP method) (dymad.modules.mlp.ResBlockMLP method) (dymad.modules.ResBlockGNN method) (dymad.modules.ResBlockMLP method) (dymad.modules.sequential.SequentialBase method) (dymad.modules.SequentialBase method) (dymad.utils.control.ControlInterpolator method) (dymad.utils.ControlInterpolator method) (dymad.utils.JaxWrapper method) (dymad.utils.wrapper.JaxMultiInFn static method) (dymad.utils.wrapper.JaxWrapper method) from_checkpoint_payload() (dymad.training.ArtifactRegistry class method) (dymad.training.phase_runtime.ArtifactRegistry class method) (dymad.training.phase_runtime.TrainerState class method) (dymad.training.TrainerState class method) from_config() (dymad.training.execution_services.ExecutionServices class method) (dymad.training.ExecutionServices class method) from_driver_config() (dymad.training.execution_services.ExecutionServices class method) (dymad.training.ExecutionServices class method) from_graph_arrays() (dymad.io.series_adapter.SeriesAdapter static method) from_regular_arrays() (dymad.io.series_adapter.SeriesAdapter static method) from_tensors() (dymad.numerics.Manifold class method) (dymad.numerics.manifold.Manifold class method) full_matrix (dymad.sako.snapshot.KoopmanWeightSnapshot attribute) FZU_MAP (in module dymad.models.components) fzu_selector() (in module dymad.models.helpers) G gamma (dymad.losses.losses.VPTLoss attribute) (dymad.losses.VPTLoss attribute) gaussian_control() (in module dymad.utils.sampling) gaussian_noise() (in module dymad.utils.sampling) gaussian_x0() (in module dymad.utils.sampling) GCL_MAP (in module dymad.modules.helpers) generate_coef() (in module dymad.numerics) (in module dymad.numerics.spectrum) generate_weak_weights() (in module dymad.numerics) (in module dymad.numerics.weak) genMat() (in module dymad.numerics.dm) get() (dymad.training.ArtifactRegistry method) (dymad.training.phase_runtime.ArtifactRegistry method) get_backward_modes() (dymad.io.checkpoint.DataInterface method) (dymad.io.DataInterface method) (dymad.sako.adapter.SpectralRuntime method) get_batch_ct() (in module dymad.training.ls_update) get_batch_dt() (in module dymad.training.ls_update) get_by_dotted_key() (in module dymad.training.helper) get_dims() (in module dymad.models) (in module dymad.models.helpers) get_forward_modes() (dymad.io.checkpoint.DataInterface method) (dymad.io.DataInterface method) (dymad.sako.adapter.SpectralRuntime method) get_length() (dymad.utils.scheduler.SweepScheduler method) get_metric() (dymad.training.phase_runtime.PhaseResult method) (dymad.training.PhaseResult method) get_weights() (dymad.sako.base.SAInterface method) GKBF (in module dymad.models.collections) GKM (in module dymad.models.collections) GLDM (in module dymad.models.collections) GLTI (in module dymad.models.collections) gmls() (dymad.numerics.Manifold method) (dymad.numerics.manifold.Manifold method) GNN (class in dymad.modules) (class in dymad.modules.gnn) GRAPH (dymad.models.model_base.ComposedDynamics attribute) (dymad.models.recipes.CD_KMM attribute) (dymad.models.recipes_corr.TemplateCorrAlg attribute) (dymad.models.recipes_corr.TemplateCorrDif attribute) graph_control (dymad.models.runtime_view.ComponentInputView property) graph_data_prep() (in module dymad.io.checkpoint) graph_mode (dymad.models.model_spec.ModelSpec attribute) (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.ModelSpec attribute) (dymad.models.ResolvedModelSpec attribute) graph_state (dymad.models.runtime_view.ComponentInputView property) grid_x0() (in module dymad.utils.sampling) H hist (dymad.training.phase_runtime.TrainingHistoryArtifact attribute) (dymad.training.TrainingHistoryArtifact attribute) I IdenCatGNN (class in dymad.modules) (class in dymad.modules.gnn) IdenCatMLP (class in dymad.modules) (class in dymad.modules.mlp) INIT_MAP_B (in module dymad.modules.helpers) INIT_MAP_W (in module dymad.modules.helpers) input_dim (dymad.sako.snapshot.SpectralSnapshot attribute) input_order (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) inv_softplus() (in module dymad.modules.kernel) is_operator_valued (dymad.modules.kernel.KernelAbstract property) (dymad.modules.kernel.KernelOperatorValued property) (dymad.modules.kernel.KernelScalarValued property) (dymad.modules.KernelAbstract property) (dymad.modules.KernelOperatorValued property) (dymad.modules.KernelScalarValued property) iter_folds() (dymad.training.driver.DriverBase method) (dymad.training.driver.KFoldDriver method) (dymad.training.driver.SingleSplitDriver method) (dymad.training.DriverBase method) (dymad.training.SingleSplitDriver method) iter_param_grid() (in module dymad.training) (in module dymad.training.helper) iter_runtime_batches() (in module dymad.training.batch_adapter) J jacobi_polynomial() (in module dymad.numerics.weak) jacobi_polynomial_derivative() (in module dymad.numerics.weak) jax_to_torch() (in module dymad.utils.wrapper) JaxMultiInFn (class in dymad.utils.wrapper) JaxWrapper (class in dymad.utils) (class in dymad.utils.wrapper) K KBF (in module dymad.models.collections) KernelAbstract (class in dymad.modules) (class in dymad.modules.kernel) KernelOperatorValued (class in dymad.modules) (class in dymad.modules.kernel) KernelOperatorValuedScalars (class in dymad.modules) (class in dymad.modules.kernel) KernelOpSeparable (class in dymad.modules) (class in dymad.modules.kernel) KernelOpTangent (class in dymad.modules) (class in dymad.modules.kernel) KernelScalarValued (class in dymad.modules) (class in dymad.modules.kernel) KernelScDM (class in dymad.modules) (class in dymad.modules.kernel) KernelScExp (class in dymad.modules) (class in dymad.modules.kernel) KernelScRBF (class in dymad.modules) (class in dymad.modules.kernel) key (dymad.training.driver.CVSearchBoundSpec attribute) keys() (dymad.training.ArtifactRegistry method) (dymad.training.phase_runtime.ArtifactRegistry method) KFoldDriver (class in dymad.training.driver) kind (dymad.models.DecoderSpec attribute) (dymad.models.DynamicsSpec attribute) (dymad.models.EncoderSpec attribute) (dymad.models.FeatureSpec attribute) (dymad.models.model_spec.DecoderSpec attribute) (dymad.models.model_spec.DynamicsSpec attribute) (dymad.models.model_spec.EncoderSpec attribute) (dymad.models.model_spec.FeatureSpec attribute) (dymad.models.model_spec.RecipeSpec attribute) (dymad.models.RecipeSpec attribute) (dymad.training.phase_runtime.PhaseRecord attribute) (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseRecord attribute) (dymad.training.PhaseResult attribute) (dymad.training.phases.AnalysisPhaseSpec attribute) (dymad.training.phases.BasePhaseSpec attribute) (dymad.training.phases.DataPhaseSpec attribute) (dymad.training.phases.ExportPhaseSpec attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) (dymad.training.phases.OptimizerPhaseSpec attribute) KM (in module dymad.models.collections) KMM (in module dymad.models.collections) koopman_weights (dymad.sako.snapshot.SpectralSnapshot attribute) KoopmanWeightSnapshot (class in dymad.sako.snapshot) KRRBase (class in dymad.modules) (class in dymad.modules.krr) KRRMultiOutputIndep (class in dymad.modules) (class in dymad.modules.krr) KRRMultiOutputShared (class in dymad.modules) (class in dymad.modules.krr) KRROperatorValued (class in dymad.modules) (class in dymad.modules.krr) KRRTangent (class in dymad.modules) (class in dymad.modules.krr) kwargs (dymad.training.LinearSolvePhaseSpec attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) L laplace_noise() (in module dymad.utils.sampling) latent_state (dymad.models.MemorySpec attribute) (dymad.models.model_spec.MemorySpec attribute) LDM (in module dymad.models.collections) LDMG (in module dymad.models.collections) left_eigvecs (dymad.sako.adapter.SpectralEigensystem attribute) left_factor (dymad.sako.snapshot.KoopmanWeightSnapshot attribute) LIN_MAP (in module dymad.models.components) linear_eval() (dymad.models.model_base.ComposedDynamics method) linear_eval_graph() (in module dymad.models.components) linear_eval_smpl() (in module dymad.models.components) linear_features() (dymad.models.model_base.ComposedDynamics method) linear_features_graph() (in module dymad.models.components) linear_features_smpl() (in module dymad.models.components) linear_mode (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.ResolvedModelSpec attribute) linear_solve() (dymad.models.model_base.ComposedDynamics method) (dymad.models.recipes.CD_KM method) (dymad.models.recipes.CD_KMM method) (dymad.models.recipes.CD_KMSK method) LinearRegressionPhase (class in dymad.training.phases) LinearSolvePhase (class in dymad.training.phases) LinearSolvePhaseSpec (class in dymad.training) (class in dymad.training.phases) LinearSolveRecord (class in dymad.training.phase_runtime) LinearSolveReportArtifact (class in dymad.training) (class in dymad.training.phase_runtime) LinearTrainer (class in dymad.training) (class in dymad.training.trainer) load_config() (in module dymad.utils) (in module dymad.utils.misc) load_data() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.trajectory_manager.TrajectoryManagerGraph method) (dymad.io.TrajectoryManager method) (dymad.io.TrajectoryManagerGraph method) load_model() (in module dymad.io) (in module dymad.io.checkpoint) load_run_checkpoint() (dymad.training.trainer_run.TrainerRun method) (dymad.training.TrainerRun method) load_state_dict() (dymad.models.recipes.CD_KM method) (dymad.models.recipes.CD_KMM method) (dymad.numerics.DM method) (dymad.numerics.dm.DM method) (dymad.numerics.dm.DMF method) (dymad.numerics.dm.VBDM method) (dymad.numerics.DMF method) (dymad.numerics.VBDM method) (dymad.utils.scheduler.SchedulerBase method) (dymad.utils.scheduler.StandardScheduler method) (dymad.utils.scheduler.SweepScheduler method) log_level (dymad.training.execution_services.ExecutionServices attribute) (dymad.training.ExecutionServices attribute) log_stdout (dymad.training.execution_services.ExecutionServices attribute) (dymad.training.ExecutionServices attribute) logger_prefix() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) logm_low_rank() (in module dymad.numerics) (in module dymad.numerics.linalg) loss (dymad.training.phase_runtime.LinearSolveRecord attribute) LOSS_MAP (in module dymad.losses.losses) lower (dymad.training.driver.CVSearchBoundSpec attribute) LSUpdater (class in dymad.training.ls_update) LTI (in module dymad.models.collections) M make_autoencoder() (in module dymad.modules) (in module dymad.modules.collections) make_kernel() (in module dymad.modules) (in module dymad.modules.collections) make_krr() (in module dymad.modules) (in module dymad.modules.collections) make_network() (in module dymad.modules) (in module dymad.modules.collections) make_random_matrix() (in module dymad.numerics) (in module dymad.numerics.linalg) make_scheduler() (in module dymad.utils) (in module dymad.utils.scheduler) Manifold (class in dymad.numerics) (class in dymad.numerics.manifold) ManifoldAltTree (class in dymad.numerics) (class in dymad.numerics.manifold) ManifoldAnalytical (class in dymad.numerics) (class in dymad.numerics.manifold) mean_metric (dymad.training.CVResult attribute) (dymad.training.helper.CVResult attribute) memory (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) MemorySpec (class in dymad.models) (class in dymad.models.model_spec) metadata (dymad.sako.snapshot.SpectralSnapshot attribute) method (dymad.training.LinearSolvePhaseSpec attribute) (dymad.training.phase_runtime.LinearSolveRecord attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) metrics (dymad.training.EvaluationArtifact attribute) (dymad.training.phase_runtime.EvaluationArtifact attribute) (dymad.training.phase_runtime.PhaseRecord attribute) (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseRecord attribute) (dymad.training.PhaseResult attribute) MLP (class in dymad.modules) (class in dymad.modules.mlp) mode (dymad.sako.snapshot.KoopmanWeightSnapshot attribute) mode_split() (in module dymad.numerics) (in module dymad.numerics.linalg) model (dymad.training.ModelArtifact attribute) (dymad.training.phase_runtime.ModelArtifact attribute) model_class (dymad.sako.snapshot.SpectralSnapshot attribute) model_cls (dymad.models.model_spec.ModelSpec property) (dymad.models.model_spec.RecipeSpec attribute) (dymad.models.ModelSpec property) (dymad.models.RecipeSpec attribute) model_ref (dymad.io.BoundaryLoadTrace attribute) (dymad.io.checkpoint.BoundaryLoadTrace attribute) model_spec (dymad.models.collections.PredefinedModel attribute) (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.PredefinedModel attribute) (dymad.models.ResolvedModelSpec attribute) ModelArtifact (class in dymad.training) (class in dymad.training.phase_runtime) ModelSpec (class in dymad.models) (class in dymad.models.model_spec) ModelSpecValidationError, [1] module dymad.io dymad.io.checkpoint dymad.io.series_adapter dymad.io.trajectory_manager dymad.losses dymad.losses.losses dymad.models dymad.models.collections dymad.models.components dymad.models.helpers dymad.models.model_base dymad.models.model_spec dymad.models.prediction dymad.models.recipes dymad.models.recipes_corr dymad.models.rollout_engine dymad.models.runtime_view dymad.modules dymad.modules.collections dymad.modules.gnn dymad.modules.helpers dymad.modules.kernel dymad.modules.krr dymad.modules.linear dymad.modules.misc dymad.modules.mlp dymad.modules.sequential dymad.numerics dymad.numerics.complex dymad.numerics.denoising dymad.numerics.dm dymad.numerics.gradients dymad.numerics.linalg dymad.numerics.manifold dymad.numerics.spectrum dymad.numerics.time_int dymad.numerics.weak dymad.sako dymad.sako.adapter dymad.sako.base dymad.sako.plotting dymad.sako.rals dymad.sako.sako dymad.sako.snapshot dymad.training dymad.training.batch_adapter dymad.training.driver dymad.training.execution_services dymad.training.helper dymad.training.ls_update dymad.training.phase_pipeline dymad.training.phase_runtime dymad.training.phases dymad.training.trainer dymad.training.trainer_run dymad.utils dymad.utils.control dymad.utils.graph dymad.utils.misc dymad.utils.plot dymad.utils.sampling dymad.utils.scheduler dymad.utils.wrapper N name (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) (dymad.training.phase_runtime.PhaseRecord attribute) (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseRecord attribute) (dymad.training.PhaseResult attribute) (dymad.training.phases.AnalysisPhaseSpec attribute) (dymad.training.phases.BasePhaseSpec attribute) (dymad.training.phases.DataPhaseSpec attribute) (dymad.training.phases.ExportPhaseSpec attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) (dymad.training.phases.OptimizerPhaseSpec attribute) nelder_mead_like_search_indices() (in module dymad.training.helper) NN_MAP (in module dymad.modules.collections) NodeOptimizerPhase (class in dymad.training.phases) NODETrainer (class in dymad.training) (class in dymad.training.trainer) normalize_phase_specs() (in module dymad.training) (in module dymad.training.phases) O obs_dim (dymad.sako.snapshot.SpectralSnapshot attribute) OneStepOptimizerPhase (class in dymad.training.phases) OneStepTrainer (class in dymad.training) (class in dymad.training.trainer) operation (dymad.training.DataPhaseSpec attribute) (dymad.training.phases.DataPhaseSpec attribute) optimizer (dymad.training.OptimizerStateArtifact attribute) (dymad.training.phase_runtime.OptimizerStateArtifact attribute) OptimizerPhaseSpec (class in dymad.training) (class in dymad.training.phases) OptimizerStateArtifact (class in dymad.training) (class in dymad.training.phase_runtime) outputs (dymad.training.ExportArtifact attribute) (dymad.training.phase_runtime.ExportArtifact attribute) owner_phase (dymad.training.OptimizerStateArtifact attribute) (dymad.training.phase_runtime.OptimizerStateArtifact attribute) P params (dymad.training.CVResult attribute) (dymad.training.helper.CVResult attribute) (dymad.training.LinearSolvePhaseSpec attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) parity (dymad.training.driver.CVSearchBoundSpec attribute) per_state_err() (in module dymad.sako.plotting) perturb_x0() (in module dymad.utils.sampling) phase_context (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseResult attribute) phase_cursor (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.TrainerState attribute) phase_name (dymad.training.phase_runtime.LinearSolveRecord attribute) phase_records (dymad.training.phase_runtime.TrainerState attribute) (dymad.training.TrainerState attribute) PhaseContext (class in dymad.training) (class in dymad.training.phase_runtime) PhasePipeline (class in dymad.training) (class in dymad.training.phase_pipeline) PhaseRecord (class in dymad.training) (class in dymad.training.phase_runtime) PhaseResult (class in dymad.training) (class in dymad.training.phase_runtime) PhaseSpecValidationError, [1] plan (dymad.io.BoundaryLoadTrace attribute) (dymad.io.checkpoint.BoundaryLoadTrace attribute) plot() (dymad.numerics.DimensionEstimator method) (dymad.numerics.manifold.DimensionEstimator method) plot2d() (dymad.numerics.Manifold method) (dymad.numerics.manifold.Manifold method) plot3d() (dymad.numerics.Manifold method) (dymad.numerics.manifold.Manifold method) plot_contour() (in module dymad.utils) (in module dymad.utils.plot) plot_cv_results() (in module dymad.utils) (in module dymad.utils.plot) plot_eigfun_2d() (dymad.sako.base.SpectralAnalysis method) (dymad.sako.plotting.SpectralPlottingAdapter method) plot_eigjac_contour() (dymad.sako.base.SpectralAnalysis method) (dymad.sako.plotting.SpectralPlottingAdapter method) plot_eigs() (dymad.sako.base.SpectralAnalysis method) (dymad.sako.plotting.SpectralPlottingAdapter method) plot_hist() (in module dymad.utils) (in module dymad.utils.plot) plot_multi_trajs() (in module dymad.utils) (in module dymad.utils.plot) plot_one_summary() (in module dymad.utils.plot) plot_one_trajectory() (in module dymad.utils.plot) plot_pred() (dymad.sako.base.SpectralAnalysis method) (dymad.sako.plotting.SpectralPlottingAdapter method) plot_summary() (in module dymad.utils) (in module dymad.utils.plot) plot_trajectory() (in module dymad.utils) (in module dymad.utils.plot) plot_vec_line() (dymad.sako.base.SpectralAnalysis method) (dymad.sako.plotting.SpectralPlottingAdapter method) precompute() (dymad.numerics.Manifold method) (dymad.numerics.manifold.Manifold method) (dymad.numerics.manifold.ManifoldAnalytical method) (dymad.numerics.ManifoldAnalytical method) PredefinedModel (class in dymad.models) (class in dymad.models.collections) predict() (dymad.models.model_base.ComposedDynamics method) (dymad.sako.base.SpectralAnalysis method) predict_continuous() (in module dymad.models.prediction) predict_continuous_exp() (in module dymad.models.prediction) predict_continuous_fenc() (in module dymad.models.prediction) predict_continuous_np() (in module dymad.models.prediction) predict_discrete() (in module dymad.models.prediction) predict_discrete_exp() (in module dymad.models.prediction) predict_krr() (dymad.numerics.dm.DMF method) (dymad.numerics.DMF method) Predictor (class in dymad.models.model_base) predictor (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.ResolvedModelSpec attribute) (dymad.models.rollout_engine.RolloutEngineSelection attribute) predictor_key (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.ResolvedModelSpec attribute) prepare_data() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.TrajectoryManager method) process_all() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.TrajectoryManager method) process_data() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.TrajectoryManager method) processor_net (dymad.models.model_spec.ResolvedModelSpec attribute) (dymad.models.recipes.RecipeResolution attribute) (dymad.models.ResolvedModelSpec attribute) projector (dymad.sako.adapter.SpectralEigensystem attribute) put() (dymad.training.ArtifactRegistry method) (dymad.training.phase_runtime.ArtifactRegistry method) R RALowRank (class in dymad.sako.rals) rals (dymad.sako.adapter.SpectralAnalysisAdapter property) randomized_svd() (in module dymad.numerics) (in module dymad.numerics.linalg) rational_kernel() (in module dymad.numerics) (in module dymad.numerics.spectrum) real_lowrank_from_eigpairs() (in module dymad.numerics) (in module dymad.numerics.linalg) recipe (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) RecipeResolution (class in dymad.models.recipes) RecipeSpec (class in dymad.models) (class in dymad.models.model_spec) record (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseResult attribute) records (dymad.training.LinearSolveReportArtifact attribute) (dymad.training.phase_runtime.LinearSolveReportArtifact attribute) replay_context() (dymad.training.phases.BasePhase method) (dymad.training.phases.ContextDataPhase method) require() (dymad.training.ArtifactRegistry method) (dymad.training.phase_runtime.ArtifactRegistry method) requires_delay_window (dymad.models.MemorySpec attribute) (dymad.models.model_spec.MemorySpec attribute) ResBlockGNN (class in dymad.modules) (class in dymad.modules.gnn) ResBlockMLP (class in dymad.modules) (class in dymad.modules.mlp) reset_optimizer (dymad.training.LinearSolvePhaseSpec attribute) (dymad.training.OptimizerPhaseSpec attribute) (dymad.training.phases.LinearSolvePhaseSpec attribute) (dymad.training.phases.OptimizerPhaseSpec attribute) resolve_recipe() (in module dymad.models.recipes) resolve_spec() (dymad.models.model_base.ComposedDynamics class method) (dymad.models.recipes.CD_KM class method) (dymad.models.recipes.CD_LDM class method) (dymad.models.recipes.CD_LFM class method) (dymad.models.recipes.CD_SDM class method) ResolvedModelSpec (class in dymad.models) (class in dymad.models.model_spec) resolvent_analysis() (dymad.sako.adapter.SpectralAnalysisAdapter method) (dymad.sako.base.SpectralAnalysis method) (in module dymad.sako.rals) results_prefix (dymad.training.execution_services.ExecutionServices attribute) (dymad.training.ExecutionServices attribute) ridge (dymad.modules.krr.KRRBase property) (dymad.modules.KRRBase property) right_eigvecs (dymad.sako.adapter.SpectralEigensystem attribute) right_factor (dymad.sako.snapshot.KoopmanWeightSnapshot attribute) rk4_step() (in module dymad.numerics) (in module dymad.numerics.time_int) rollout (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) RolloutEngineSelection (class in dymad.models.rollout_engine) RolloutSpec (class in dymad.models) (class in dymad.models.model_spec) run() (dymad.training.phase_pipeline.PhasePipeline method) (dymad.training.PhasePipeline method) (dymad.training.trainer_run.TrainerRun method) (dymad.training.TrainerRun method) run_checkpoint_path (dymad.training.trainer_run.TrainerRun property) (dymad.training.TrainerRun property) run_cv_single() (in module dymad.training.driver) RunCheckpointExportPhase (class in dymad.training.phases) runtime (dymad.models.runtime_view.ComponentInputView attribute) S SAInterface (class in dymad.sako.base) SAKO (class in dymad.sako) (class in dymad.sako.sako) sako (dymad.sako.adapter.SpectralAnalysisAdapter property) sample_count (dymad.sako.snapshot.SpectralSnapshot attribute) sanity_check() (dymad.numerics.DimensionEstimator method) (dymad.numerics.manifold.DimensionEstimator method) save_run_checkpoint() (dymad.training.trainer_run.TrainerRun method) (dymad.training.TrainerRun method) scaled_cdist() (in module dymad.modules) (in module dymad.modules.kernel) scaled_eig() (in module dymad.numerics) (in module dymad.numerics.linalg) SchedulerBase (class in dymad.utils.scheduler) schedulers (dymad.training.OptimizerStateArtifact attribute) (dymad.training.phase_runtime.OptimizerStateArtifact attribute) scl (dymad.losses.losses.VPTLoss attribute) (dymad.losses.VPTLoss attribute) select_best_cv_result() (in module dymad.training) (in module dymad.training.helper) select_rollout_engine() (in module dymad.models.rollout_engine) selection_combo_indices (dymad.training.driver.CVSearchRunResult attribute) SequentialBase (class in dymad.modules) (class in dymad.modules.sequential) SeriesAdapter (class in dymad.io.series_adapter) set_by_dotted_key() (in module dymad.training) (in module dymad.training.helper) set_conj_map() (dymad.sako.base.SpectralAnalysis method) set_data_index() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.TrajectoryManager method) set_full() (dymad.modules.FlexLinear method) (dymad.modules.linear.FlexLinear method) set_linear_weights() (dymad.models.model_base.ComposedDynamics method) set_lora() (dymad.modules.FlexLinear method) (dymad.modules.linear.FlexLinear method) set_manifold() (dymad.modules.kernel.KernelOpTangent method) (dymad.modules.KernelOpTangent method) (dymad.modules.krr.KRRTangent method) (dymad.modules.KRRTangent method) set_reference_data() (dymad.modules.kernel.KernelAbstract method) (dymad.modules.kernel.KernelOperatorValuedScalars method) (dymad.modules.kernel.KernelOpTangent method) (dymad.modules.kernel.KernelScDM method) (dymad.modules.kernel.KernelScRBF method) (dymad.modules.KernelAbstract method) (dymad.modules.KernelOperatorValuedScalars method) (dymad.modules.KernelOpTangent method) (dymad.modules.KernelScDM method) (dymad.modules.KernelScRBF method) set_train_data() (dymad.modules.krr.KRRBase method) (dymad.modules.KRRBase method) set_transforms() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.trajectory_manager.TrajectoryManagerGraph method) (dymad.io.TrajectoryManager method) (dymad.io.TrajectoryManagerGraph method) set_weights() (dymad.models.model_base.SupportsSetWeights method) (dymad.modules.FlexLinear method) (dymad.modules.linear.FlexLinear method) setup_logging() (in module dymad.utils) (in module dymad.utils.misc) SimpleRNN (class in dymad.modules) (class in dymad.modules.sequential) sine_control() (in module dymad.utils.sampling) SingleSplitDriver (class in dymad.training) (class in dymad.training.driver) snapshot (dymad.sako.base.SAInterface property) SOL_MAP (in module dymad.training.ls_update) solve_eig() (dymad.sako.SAKO method) (dymad.sako.sako.SAKO method) source (dymad.models.rollout_engine.RolloutEngineSelection attribute) SpectralAnalysis (class in dymad.sako.base) SpectralAnalysisAdapter (class in dymad.sako.adapter) SpectralEigensystem (class in dymad.sako.adapter) SpectralPlottingAdapter (class in dymad.sako.plotting) SpectralRuntime (class in dymad.sako.adapter) SpectralSnapshot (class in dymad.sako.snapshot) sphere_control() (in module dymad.utils.sampling) split (dymad.training.AnalysisPhaseSpec attribute) (dymad.training.EvaluationArtifact attribute) (dymad.training.phase_runtime.EvaluationArtifact attribute) (dymad.training.phases.AnalysisPhaseSpec attribute) StackedTrainer (class in dymad.training) (class in dymad.training.trainer) StandardScheduler (class in dymad.utils.scheduler) started_epoch (dymad.training.phase_runtime.PhaseRecord attribute) (dymad.training.PhaseRecord attribute) state (dymad.models.runtime_view.ComponentInputView property) state_dict() (dymad.numerics.DM method) (dymad.numerics.dm.DM method) (dymad.numerics.dm.DMF method) (dymad.numerics.dm.VBDM method) (dymad.numerics.DMF method) (dymad.numerics.VBDM method) (dymad.utils.scheduler.SchedulerBase method) (dymad.utils.scheduler.StandardScheduler method) (dymad.utils.scheduler.SweepScheduler method) std_metric (dymad.training.CVResult attribute) (dymad.training.helper.CVResult attribute) step() (dymad.utils.scheduler.SchedulerBase method) (dymad.utils.scheduler.StandardScheduler method) (dymad.utils.scheduler.SweepScheduler method) StepwiseModel (class in dymad.modules) (class in dymad.modules.sequential) student_t_noise() (in module dymad.utils.sampling) SummaryExportPhase (class in dymad.training.phases) supports_control_inputs (dymad.models.model_spec.RolloutSpec attribute) (dymad.models.RolloutSpec attribute) SupportsSetWeights (class in dymad.models.model_base) SweepScheduler (class in dymad.utils.scheduler) T t (dymad.modules.kernel.KernelScDM property) (dymad.modules.KernelScDM property) (dymad.utils.control.ControlInterpolator attribute) (dymad.utils.ControlInterpolator attribute) TakeFirst (class in dymad.modules.misc) TakeFirstGraph (class in dymad.modules.misc) tangent_1circle() (in module dymad.numerics) (in module dymad.numerics.manifold) tangent_2torus() (in module dymad.numerics) (in module dymad.numerics.manifold) TemplateCorrAlg (class in dymad.models.recipes_corr) TemplateCorrDif (class in dymad.models.recipes_corr) time_domain (dymad.models.model_spec.ModelSpec attribute) (dymad.models.ModelSpec attribute) to_tensors() (dymad.numerics.Manifold method) (dymad.numerics.manifold.Manifold method) torch_jacobian() (in module dymad.numerics) (in module dymad.numerics.gradients) torch_to_jax() (in module dymad.utils.wrapper) train() (dymad.training.driver.DriverBase method) (dymad.training.DriverBase method) train_loader (dymad.training.phase_runtime.PhaseContext attribute) (dymad.training.PhaseContext attribute) train_md (dymad.training.ModelArtifact attribute) (dymad.training.phase_runtime.ModelArtifact attribute) (dymad.training.phase_runtime.PhaseContext attribute) (dymad.training.PhaseContext attribute) train_set (dymad.training.phase_runtime.PhaseContext attribute) (dymad.training.PhaseContext attribute) train_sets (dymad.training.driver.DriverBase attribute) trainer (dymad.training.OptimizerPhaseSpec attribute) (dymad.training.phases.OptimizerPhaseSpec attribute) trainer_state (dymad.training.phase_runtime.PhaseResult attribute) (dymad.training.PhaseResult attribute) TrainerRun (class in dymad.training) (class in dymad.training.trainer_run) TrainerState (class in dymad.training) (class in dymad.training.phase_runtime) training (dymad.losses.losses.VPTLoss attribute) (dymad.losses.losses.WMSELoss attribute) (dymad.utils.control.ControlInterpolator attribute) TrainingCheckpointError, [1] TrainingHistoryArtifact (class in dymad.training) (class in dymad.training.phase_runtime) TrajectoryManager (class in dymad.io) (class in dymad.io.trajectory_manager) TrajectoryManagerGraph (class in dymad.io) (class in dymad.io.trajectory_manager) TrajectorySampler (class in dymad.utils) (class in dymad.utils.sampling) transform() (dymad.numerics.DM method) (dymad.numerics.dm.DM method) (dymad.numerics.dm.DMF method) (dymad.numerics.dm.VBDM method) (dymad.numerics.DMF method) (dymad.numerics.VBDM method) transform_naive() (dymad.numerics.dm.VBDM method) (dymad.numerics.VBDM method) truncate_sequence() (in module dymad.numerics) (in module dymad.numerics.linalg) truncated_lstsq() (in module dymad.numerics.linalg) truncated_svd() (in module dymad.numerics) (in module dymad.numerics.linalg) typed_spec() (dymad.models.collections.PredefinedModel method) (dymad.models.PredefinedModel method) U u (dymad.utils.control.ControlInterpolator attribute) (dymad.utils.ControlInterpolator attribute) unflatten_nodes() (dymad.models.runtime_view.ComponentInputView method) uniform_noise() (in module dymad.utils.sampling) uniform_x0() (in module dymad.utils.sampling) update() (dymad.training.ls_update.LSUpdater method) update_config() (dymad.io.trajectory_manager.TrajectoryManager method) (dymad.io.TrajectoryManager method) updated_parameters (dymad.training.phase_runtime.LinearSolveRecord attribute) upper (dymad.training.driver.CVSearchBoundSpec attribute) V valid_loader (dymad.training.phase_runtime.PhaseContext attribute) (dymad.training.PhaseContext attribute) valid_md (dymad.training.ModelArtifact attribute) (dymad.training.phase_runtime.ModelArtifact attribute) (dymad.training.phase_runtime.PhaseContext attribute) (dymad.training.PhaseContext attribute) valid_set (dymad.training.phase_runtime.PhaseContext attribute) (dymad.training.PhaseContext attribute) valid_sets (dymad.training.driver.DriverBase attribute) ValidationAnalysisPhase (class in dymad.training.phases) value_kind (dymad.training.driver.CVSearchBoundSpec attribute) VanillaRNN (class in dymad.modules) (class in dymad.modules.sequential) VBDM (class in dymad.numerics) (class in dymad.numerics.dm) visualize_model() (in module dymad.io) (in module dymad.io.checkpoint) vpt_loss() (in module dymad.losses) (in module dymad.losses.losses) VPTLoss (class in dymad.losses) (class in dymad.losses.losses) W WeakFormOptimizerPhase (class in dymad.training.phases) WeakFormTrainer (class in dymad.training) (class in dymad.training.trainer) with_device() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) with_paths() (dymad.training.execution_services.ExecutionServices method) (dymad.training.ExecutionServices method) wmse_loss() (in module dymad.losses) (in module dymad.losses.losses) WMSELoss (class in dymad.losses) (class in dymad.losses.losses) X X0_MAP (in module dymad.utils.sampling) Z zu_blin_no_const() (in module dymad.models.components) zu_blin_no_const_graph() (in module dymad.models.components) zu_blin_with_const() (in module dymad.models.components) zu_blin_with_const_graph() (in module dymad.models.components) zu_cat_none() (in module dymad.models.components) zu_cat_smpl() (in module dymad.models.components) zu_cat_smpl_graph() (in module dymad.models.components)