Alternatives and similar repositories for ssm-jax:

Users that are interested in ssm-jax are comparing it to the libraries listed below

• dieterichlawson / snax
  ☆12Updated last year
• catniplab / tree_structured_rslds
  Tree-structured recurrent switching linear dynamical systems
  ☆36Updated 4 years ago
• mackelab / delfi
  Density estimation likelihood-free inference. No longer actively developed see https://github.com/mackelab/sbi instead
  ☆73Updated 4 years ago
• tachukao / mgplvm-pytorch
  ☆23Updated last year
• acerbilab / pyvbmc
  PyVBMC: Variational Bayesian Monte Carlo algorithm for posterior and model inference in Python
  ☆117Updated 2 weeks ago
• vruetten / GPFADS
  Gaussian Process Factor Analysis with Dynamical Structure
  ☆16Updated 4 years ago
• DurstewitzLab / GTF-shPLRNN
  ☆13Updated 10 months ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆100Updated last year
• slinderman / recurrent-slds
  Recurrent Switching Linear Dynamical Systems
  ☆105Updated last year
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆98Updated last year
• wesselb / gpar
  Implementation of the Gaussian Process Autoregressive Regression Model
  ☆62Updated last month
• wesselb / lab
  A generic interface for linear algebra backends
  ☆71Updated 7 months ago
• marineschimel / ilqr_vae
  ☆11Updated 2 years ago
• EEA-sensors / sequential-parallelization-examples
  This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.
  ☆31Updated last year
• BouchardLab / DynamicalComponentsAnalysis
  Dynamical Components Analysis
  ☆31Updated last year
• andrew-cr / online_var_fil
  Code for our paper: Online Variational Filtering and Parameter Learning
  ☆18Updated 3 years ago
• gatsby-sahani / gp-sde
  ☆13Updated last year
• earcher / vilds
  Black box variational inference for state space models
  ☆1Updated 8 years ago
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• davidzoltowski / ssmdm
  Recurrent state-space models for decision making
  ☆30Updated 2 years ago
• petergchang / sarkka-jax
  ☆11Updated 2 years ago
• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆30Updated last month
• blei-lab / variational-smc
  Reference implementation of variational sequential Monte Carlo proposed by Naesseth et al. "Variational Sequential Monte Carlo" (2018)
  ☆64Updated 5 years ago
• lindermanlab / SVAE-Revisited
  ☆9Updated last year
• probml / rebayes
  Recursive Bayesian Estimation (Sequential / Online Inference)
  ☆58Updated 10 months ago
• EEA-sensors / sqrt-parallel-smoothers
  A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…
  ☆12Updated 2 months ago
• gpapamak / snl
  Sequential Neural Likelihood
  ☆39Updated 5 years ago
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 3 years ago
• jimmysmith1919 / JSLDS_public
  ☆12Updated 3 years ago