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
  ☆58Updated 10 months 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
• wrongu / repsim
  PyTorch-based library for various kinds of representational-similarity analysis
  ☆24Updated 10 months ago
• conorheins / bayesian-mechanics-sdes
  ☆13Updated 2 years ago
• tachukao / mgplvm-pytorch
  ☆23Updated last year
• earcher / vilds
  Black box variational inference for state space models
  ☆1Updated 8 years ago
• vruetten / GPFADS
  Gaussian Process Factor Analysis with Dynamical Structure
  ☆16Updated 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
• fzenke / randman
  ☆13Updated 2 years ago
• catniplab / tree_structured_rslds
  Tree-structured recurrent switching linear dynamical systems
  ☆36Updated 4 years ago
• AdrienCorenflos / jax-workshop
  JAX workshop for beginners
  ☆8Updated 3 months ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated last month
• NeilGirdhar / efax
  Exponential families for JAX
  ☆66Updated this week
• andrew-cr / online_var_fil
  Code for our paper: Online Variational Filtering and Parameter Learning
  ☆18Updated 3 years ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆99Updated last year
• 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 4 months ago
• baumgach / tue-slurm-helloworld
  Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container
  ☆33Updated 2 years ago
• lindermanlab / SVAE-Revisited
  ☆9Updated last year
• petergchang / sarkka-jax
  ☆11Updated 2 years ago
• tripdancer0916 / pytorch-fixed-point-analysis
  Implementation of fixed point analysis for Recurrent Neural Network.
  ☆20Updated 5 years ago
• marineschimel / ilqr_vae
  ☆11Updated 2 years ago
• BouchardLab / DynamicalComponentsAnalysis
  Dynamical Components Analysis
  ☆32Updated last year
• mattgolub / fixed-point-finder
  FixedPointFinder: A Tensorflow toolbox for identifying and characterizing fixed points in recurrent neural networks
  ☆96Updated 8 months ago
• mackelab / simformer
  Combination of transformers and diffusion models for flexible all-in-one simulation-based inference
  ☆63Updated 10 months ago
• lindermanlab / PPSeq.jl
  Neyman-Scott point process model to identify sequential firing patterns in high-dimensional spike trains
  ☆67Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated this week
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 3 years ago
• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆30Updated 3 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated 2 weeks ago
• slinderman / thesis
  My PhD Thesis
  ☆22Updated 8 years ago