Alternatives and similar repositories for snax

Users that are interested in snax are comparing it to the libraries listed below

• lindermanlab / ssm-jax
  Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
  ☆61Updated last year
• tachukao / mgplvm-pytorch
  ☆24Updated 2 years ago
• lindermanlab / gpslds
  Project repo for gpSLDS
  ☆15Updated 11 months ago
• vruetten / GPFADS
  Gaussian Process Factor Analysis with Dynamical Structure
  ☆17Updated 5 years ago
• petergchang / sarkka-jax
  ☆11Updated 3 years ago
• marineschimel / ilqr_vae
  ☆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
• wesselb / lab
  A generic interface for linear algebra backends
  ☆75Updated 3 weeks ago
• lindermanlab / SVAE-Revisited
  ☆10Updated last year
• wrongu / repsim
  PyTorch-based library for various kinds of representational-similarity analysis
  ☆24Updated last year
• mackelab / simformer
  Combination of transformers and diffusion models for flexible all-in-one simulation-based inference
  ☆75Updated 5 months ago
• jaxleyverse / jaxley
  Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.
  ☆152Updated this week
• conorheins / bayesian-mechanics-sdes
  ☆14Updated 3 years ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆106Updated 10 months ago
• acerbilab / pyvbmc
  PyVBMC: Variational Bayesian Monte Carlo algorithm for posterior and model inference in Python
  ☆121Updated last week
• NeilGirdhar / efax
  Exponential families for JAX
  ☆75Updated last month
• mackelab / neuralgbi
  Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…
  ☆11Updated 2 years 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 last year
• RothkopfLab / lqg
  Inverse optimal control for continuous psychophysics
  ☆29Updated last week
• anirgalgali / residual-dynamics
  Code for Galgali et al, 2023
  ☆14Updated 2 years ago
• tripdancer0916 / pytorch-fixed-point-analysis
  Implementation of fixed point analysis for Recurrent Neural Network.
  ☆23Updated 5 years ago
• mbaltieri / GeneralisedFiltering
  General framework for Bayesian inversion of continuous hierarchical models
  ☆10Updated 4 years ago
• acerbilab / pybads
  PyBADS: Bayesian Adaptive Direct Search optimization algorithm for model fitting in Python
  ☆79Updated last month
• catniplab / tree_structured_rslds
  Tree-structured recurrent switching linear dynamical systems
  ☆37Updated 5 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated 2 years ago
• BouchardLab / DynamicalComponentsAnalysis
  Dynamical Components Analysis
  ☆33Updated 3 months ago
• KietzmannLab / EmergentPredictiveCoding
  ☆40Updated last year
• fzenke / randman
  ☆13Updated 3 years ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated 3 weeks ago
• mattgolub / fixed-point-finder
  FixedPointFinder: A Tensorflow toolbox for identifying and characterizing fixed points in recurrent neural networks
  ☆104Updated last year