Model Documentation

These pages are generated from registered model config classes, model class docstrings, and config field defaults.

After editing a model docstring or config, regenerate these files with:

python scripts/generate_model_docs.py
Model Registry Name Summary
BGPFA bgpfa Variational Bayesian GPFA with ARD and differentiable ELBO training.
CASSM cassm Thin wrapper around the bundled CASSM sparse filter/smoother.
GPFA gpfa Gaussian Process Factor Analysis with diagonal observation noise.
ILQRVAE ilqr_vae Optimization-based iLQR-VAE with posterior-control inference and ELBO training.
Kalman kalman Dense Kalman filter baseline adapted from the CASSM source.
LangevinFlow langevin_flow LangevinFlow sequential VAE for binned neural spike counts.
LFADS lfads Latent Factor Analysis via Dynamical Systems for binned spike counts.
MINT mint Mesh of Idealized Neural Trajectories adapted to the LaDyS API.
NDT ndt Transformer encoder trained with a masked Poisson spike objective.
PSTH psth Peri-stimulus time histogram baseline.
Smoothing smoothing Gaussian-smoothed spike-count baseline.
STNDT stndt Spatiotemporal Neural Data Transformer for binned spike counts.