Alternatives and similar repositories for ssm-jax

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

• acerbilab / pyvbmc
  PyVBMC: Variational Bayesian Monte Carlo algorithm for posterior and model inference in Python
  ☆120Updated 5 months ago
• dieterichlawson / snax
  ☆12Updated 2 years ago
• mackelab / delfi
  Density estimation likelihood-free inference. No longer actively developed see https://github.com/mackelab/sbi instead
  ☆73Updated 5 years ago
• AdrienCorenflos / jax-workshop
  JAX workshop for beginners
  ☆8Updated 6 months ago
• EEA-sensors / sequential-parallelization-examples
  This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.
  ☆32Updated last year
• tachukao / mgplvm-pytorch
  ☆23Updated last year
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆104Updated last year
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 4 months ago
• catniplab / tree_structured_rslds
  Tree-structured recurrent switching linear dynamical systems
  ☆36Updated 5 years ago
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• slinderman / recurrent-slds
  Recurrent Switching Linear Dynamical Systems
  ☆111Updated 2 years ago
• acerbilab / pybads
  PyBADS: Bayesian Adaptive Direct Search optimization algorithm for model fitting in Python
  ☆75Updated 5 months ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆72Updated 3 weeks ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆97Updated 5 months ago
• probml / rebayes
  Recursive Bayesian Estimation (Sequential / Online Inference)
  ☆59Updated last year
• RothkopfLab / lqg
  Inverse optimal control for continuous psychophysics
  ☆27Updated 2 weeks ago
• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆31Updated 5 months ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆100Updated 2 years ago
• earcher / vilds
  Black box variational inference for state space models
  ☆2Updated 9 years ago
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 4 years ago
• BouchardLab / DynamicalComponentsAnalysis
  Dynamical Components Analysis
  ☆32Updated 2 years ago
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆75Updated this week
• vruetten / GPFADS
  Gaussian Process Factor Analysis with Dynamical Structure
  ☆16Updated 4 years ago
• jaxleyverse / jaxley
  Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
  ☆104Updated last week
• mbaltieri / GeneralisedFiltering
  General framework for Bayesian inversion of continuous hierarchical models
  ☆10Updated 3 years ago
• canyon289 / ssm_book_club
  ☆17Updated 2 years ago
• wesselb / gpar
  Implementation of the Gaussian Process Autoregressive Regression Model
  ☆67Updated 5 months ago
• mattjj / pyslds
  ☆96Updated 7 years ago
• arsedler9 / nlb-lightning
  PyTorch Lightning utilities that make it easier to train and evaluate deep models for the Neural Latents Benchmark.
  ☆8Updated 2 years ago
• f-dangel / phd-thesis
  Source code for my PhD thesis: Backpropagation Beyond the Gradient
  ☆20Updated 2 years ago