Index

A

AnalyticDynJacMixin (class in ceem.dynamics) AnalyticObsJacMixin (class in ceem.dynamics)

apply_inds() (ceem.opt_criteria.GaussianDynamicsCriterion method) (ceem.opt_criteria.GaussianObservationCriterion method)

B

BasicGroupSOSCriterion (class in ceem.opt_criteria) batched_forward() (ceem.opt_criteria.Criterion method)

batched_sample_forward() (ceem.opt_criteria.Criterion method) BlockSparseGroupSOSCriterion (class in ceem.opt_criteria)

C

C2DSystem (class in ceem.dynamics) ceem module CEEM (class in ceem.ceem) ceem.baseline_utils module ceem.ceem module ceem.data_utils module ceem.dynamics module ceem.exp_utils module ceem.learner module ceem.logger module ceem.nested module ceem.nn module ceem.odesolver module ceem.opt_criteria module

ceem.particleem module ceem.smoother module ceem.systems module ceem.systems.discretelinear module ceem.systems.lorenzattractor module ceem.systems.lorenzsystem module ceem.systems.maglev module ceem.systems.springmassdamper module ceem.utils module choice() (in module ceem.smoother) compute_mean() (ceem.particleem.faPF method) compute_rms() (in module ceem.exp_utils) ContinuousDynamicalSystem (class in ceem.dynamics) Criterion (class in ceem.opt_criteria) critical() (in module ceem.logger)

D

debug() (in module ceem.logger) default_lorenz_attractor() (in module ceem.systems.lorenzattractor) default_lorenz_system() (in module ceem.systems.lorenzsystem) DEFAULT_NN_KWARGS() (in module ceem.baseline_utils) disable_grad() (in module ceem.utils) DiscreteDynamicalSystem (class in ceem.dynamics)

DiscreteLinear (class in ceem.systems.discretelinear) dist() (ceem.particleem.Resampler method) dynamics() (ceem.dynamics.KinoDynamicalSystem method) (ceem.systems.springmassdamper.SpringMassDamper method) DynamicsModule (class in ceem.systems.discretelinear) DynJacMixin (class in ceem.dynamics)

E

EKF() (in module ceem.smoother) ensure_default_torch_kwargs() (in module ceem.learner)

error() (in module ceem.logger) Euler (class in ceem.odesolver) exception() (in module ceem.logger)

F

fa_loglikelihood() (ceem.particleem.faPF method) faPF (class in ceem.particleem) FFBSi() (ceem.particleem.faPF method) FFBSm() (ceem.particleem.faPF method) filter() (ceem.particleem.faPF method) (ceem.smoother.ParticleSmoother method) (in module ceem.nested) filter_() (in module ceem.nested) flatten() (in module ceem.nested)

flatten_() (in module ceem.nested) forward() (ceem.baseline_utils.LagModel method) (ceem.nn.LNMLP method) (ceem.opt_criteria.Criterion method) (ceem.opt_criteria.GroupCriterion method) (ceem.opt_criteria.SOSCriterion method) (ceem.opt_criteria.STRParamCriterion method) (ceem.systems.discretelinear.DynamicsModule method) (ceem.systems.discretelinear.ObservationModule method)

G

GaussianDynamicsCriterion (class in ceem.opt_criteria) GaussianObservationCriterion (class in ceem.opt_criteria) GaussianX0Criterion (class in ceem.opt_criteria) gen_ypred_benchmark() (in module ceem.exp_utils) gen_ypred_model() (in module ceem.exp_utils)

get_grad_norm() (in module ceem.utils) get_resampling_distribution() (ceem.particleem.faPF method) get_sampling_distribution() (ceem.particleem.faPF method) get_smooth_mean() (ceem.smoother.ParticleSmoother method) GroupCriterion (class in ceem.opt_criteria) GroupSOSCriterion (class in ceem.opt_criteria)

H

HarmonicDecayScheduler() (in module ceem.particleem)

I

icdf() (ceem.particleem.Resampler method)

info() (in module ceem.logger) init_normc_() (in module ceem.nn)

J

jac_obs_theta() (ceem.dynamics.AnalyticObsJacMixin method) (ceem.dynamics.ObsJacMixin method) (ceem.systems.lorenzattractor.LorenzAttractor method) (ceem.systems.lorenzsystem.LorenzSystem method) (ceem.systems.springmassdamper.SpringMassDamper method) jac_obs_u() (ceem.dynamics.ObsJacMixin method) jac_obs_x() (ceem.dynamics.AnalyticObsJacMixin method) (ceem.dynamics.ObsJacMixin method) (ceem.systems.lorenzattractor.LorenzAttractor method) (ceem.systems.lorenzsystem.LorenzSystem method) (ceem.systems.maglev.MagLev method) (ceem.systems.springmassdamper.SpringMassDamper method) jac_resid_x() (ceem.opt_criteria.BlockSparseGroupSOSCriterion method) (ceem.opt_criteria.GaussianDynamicsCriterion method) (ceem.opt_criteria.GaussianObservationCriterion method) (ceem.opt_criteria.GaussianX0Criterion method) (ceem.opt_criteria.GroupSOSCriterion method) (ceem.opt_criteria.SOSCriterion method) (ceem.opt_criteria.STRStateCriterion method)

jac_step_theta() (ceem.dynamics.AnalyticDynJacMixin method) (ceem.dynamics.DynJacMixin method) (ceem.systems.discretelinear.DiscreteLinear method) jac_step_u() (ceem.dynamics.DynJacMixin method) jac_step_x() (ceem.dynamics.AnalyticDynJacMixin method) (ceem.dynamics.DynJacMixin method) (ceem.systems.discretelinear.DiscreteLinear method) jac_theta() (ceem.opt_criteria.Criterion method) jac_x() (ceem.opt_criteria.Criterion method)

K

kinematics() (ceem.dynamics.KinoDynamicalSystem method) (ceem.systems.springmassdamper.SpringMassDamper method)

KinoDynamicalSystem (class in ceem.dynamics)

L

LagModel (class in ceem.baseline_utils) learner() (in module ceem.learner) LNMLP (class in ceem.nn) load_helidata() (in module ceem.data_utils)

load_statistics() (in module ceem.data_utils) log_kv_or_listkv() (in module ceem.ceem) LorenzAttractor (class in ceem.systems.lorenzattractor) LorenzSystem (class in ceem.systems.lorenzsystem)

M

MagLev (class in ceem.systems.maglev) main() (in module ceem.systems.lorenzattractor) (in module ceem.systems.springmassdamper) map() (in module ceem.nested) map_() (in module ceem.nested) Midpoint (class in ceem.odesolver) module ceem ceem.baseline_utils ceem.ceem ceem.data_utils ceem.dynamics ceem.exp_utils ceem.learner ceem.logger ceem.nested ceem.nn ceem.odesolver ceem.opt_criteria ceem.particleem ceem.smoother ceem.systems ceem.systems.discretelinear ceem.systems.lorenzattractor ceem.systems.lorenzsystem ceem.systems.maglev ceem.systems.springmassdamper ceem.utils

multinomial() (ceem.particleem.Resampler method)

N

NLSsmoother() (in module ceem.smoother)

O

objectview (class in ceem.utils) ObservationModule (class in ceem.systems.discretelinear) observe() (ceem.dynamics.ContinuousDynamicalSystem method) (ceem.dynamics.DiscreteDynamicalSystem method) (ceem.systems.discretelinear.DiscreteLinear method) (ceem.systems.lorenzattractor.LorenzAttractor method) (ceem.systems.lorenzsystem.LorenzSystem method) (ceem.systems.maglev.MagLev method) (ceem.systems.springmassdamper.SpringMassDamper method)

ObsJacMixin (class in ceem.dynamics) obsll() (ceem.smoother.ParticleSmootherSystemWrapper method) odestep() (in module ceem.odesolver) order() (ceem.odesolver.Euler property) (ceem.odesolver.Midpoint property) (ceem.odesolver.RK4 property)

P

ParticleSmoother (class in ceem.smoother)

ParticleSmootherSystemWrapper (class in ceem.smoother) peak_memory_mb() (in module ceem.utils)

Q

Q_JSD_marginals() (ceem.particleem.faPF method)

Q_MCEM() (ceem.particleem.faPF method) qdim() (ceem.dynamics.KinoDynamicalSystem property)

R

recursive_Q() (ceem.particleem.SAEMTrainer method) Resampler (class in ceem.particleem) reset_params() (ceem.nn.LNMLP method) residuals() (ceem.opt_criteria.BlockSparseGroupSOSCriterion method) (ceem.opt_criteria.GaussianDynamicsCriterion method) (ceem.opt_criteria.GaussianObservationCriterion method) (ceem.opt_criteria.GaussianX0Criterion method) (ceem.opt_criteria.GroupSOSCriterion method) (ceem.opt_criteria.SOSCriterion method) (ceem.opt_criteria.STRStateCriterion method)

RK4 (class in ceem.odesolver) rk4_alt_step_func() (in module ceem.odesolver) run() (ceem.smoother.ParticleSmoother method)

S

SAEMTrainer (class in ceem.particleem) sample_initial() (ceem.particleem.faPF method) scaled_jac_x_diag() (ceem.opt_criteria.GaussianDynamicsCriterion method) (ceem.opt_criteria.GaussianObservationCriterion method) (ceem.opt_criteria.SOSCriterion method) (ceem.opt_criteria.STRStateCriterion method) scipy_minimize() (in module ceem.learner) scipyoptimizer() (in module ceem.particleem) scoped_setup (class in ceem.logger) set_rng_seed() (in module ceem.utils) setLevel() (in module ceem.logger) setup() (in module ceem.logger) smooth() (ceem.ceem.CEEM method) smoother() (ceem.smoother.ParticleSmoother method) SOSCriterion (class in ceem.opt_criteria) SpringMassDamper (class in ceem.systems.springmassdamper)

step() (ceem.ceem.CEEM method) (ceem.dynamics.C2DSystem method) (ceem.dynamics.DiscreteDynamicalSystem method) (ceem.systems.discretelinear.DiscreteLinear method) step_derivs() (ceem.dynamics.ContinuousDynamicalSystem method) (ceem.dynamics.KinoDynamicalSystem method) (ceem.systems.lorenzattractor.LorenzAttractor method) (ceem.systems.lorenzsystem.LorenzSystem method) (ceem.systems.maglev.MagLev method) step_func() (ceem.odesolver.Euler static method) (ceem.odesolver.Midpoint static method) (ceem.odesolver.RK4 static method) stratified() (ceem.particleem.Resampler method) STRParamCriterion (class in ceem.opt_criteria) STRStateCriterion (class in ceem.opt_criteria) systematic() (ceem.particleem.Resampler method)

T

temp_disable_grad() (in module ceem.utils) temp_require_grad() (in module ceem.utils) Timer (class in ceem.utils)

torch_minimize() (in module ceem.learner) torchoptimizer() (in module ceem.particleem) train() (ceem.ceem.CEEM method) (ceem.particleem.SAEMTrainer method)

U

udim() (ceem.dynamics.ContinuousDynamicalSystem property) (ceem.dynamics.DiscreteDynamicalSystem property)

V

vdim() (ceem.dynamics.KinoDynamicalSystem property)

W

warn() (in module ceem.logger)

warning() (in module ceem.logger) weights_init_mlp() (in module ceem.nn)

X

xdim() (ceem.dynamics.ContinuousDynamicalSystem property) (ceem.dynamics.DiscreteDynamicalSystem property) (ceem.dynamics.KinoDynamicalSystem property)

Y

ydim() (ceem.dynamics.ContinuousDynamicalSystem property) (ceem.dynamics.DiscreteDynamicalSystem property)

Z

zip() (in module ceem.nested)

zip_() (in module ceem.nested)