Alternatives and similar repositories for ssm-book

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

• normal-computing / posteriors
  Uncertainty quantification with PyTorch
  ☆379Updated 2 weeks ago
• probabilisticai / probai-2022
  Materials of the Nordic Probabilistic AI School 2022.
  ☆181Updated 3 years ago
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆167Updated 8 months ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆164Updated last year
• Atcold / Energy-Book
  Deep Learning, an Energy Approach
  ☆242Updated 8 months ago
• nocotan / awesome-information-geometry
  About A collection of AWESOME things about information geometry Topics
  ☆180Updated last year
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆282Updated 7 months ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• BlackHC / neural_net_checklist
  ☆215Updated last year
• philipphennig / NumericsOfML
  Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen
  ☆226Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆190Updated last month
• probml / dynamax
  A Python package for probabilistic state space modeling with JAX
  ☆924Updated last month
• probml / rebayes
  Recursive Bayesian Estimation (Sequential / Online Inference)
  ☆61Updated last year
• google-deepmind / nanodo
  ☆291Updated last year
• wilson-labs / cola
  Compositional Linear Algebra
  ☆507Updated 6 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆110Updated last year
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆300Updated last year
• probml / sts-jax
  Structural Time Series in JAX
  ☆214Updated last year
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆222Updated last year
• probabilisticai / probai-2023
  Materials of the Nordic Probabilistic AI School 2023.
  ☆91Updated 2 years ago
• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆175Updated 2 years ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆77Updated 2 years ago
• apple / ml-pararnn
  ☆168Updated 3 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆93Updated 2 years ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆406Updated this week
• marin-community / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆218Updated 3 months ago
• lindermanlab / S5
  ☆314Updated last year
• google-deepmind / synjax
  ☆251Updated 7 months ago
• AndreasBergmeister / pjax
  A projection-based framework for gradient-free and parallel learning
  ☆110Updated 7 months ago