probml / ssm-book
Interactive textbook on state-space models
☆186Updated last year
Alternatives and similar repositories for ssm-book:
Users that are interested in ssm-book are comparing it to the libraries listed below
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆201Updated last year
- Uncertainty quantification with PyTorch☆348Updated this week
- About A collection of AWESOME things about information geometry Topics☆157Updated 9 months ago
- A Python package for probabilistic state space modeling with JAX☆802Updated 2 weeks ago
- Structural Time Series in JAX☆189Updated 11 months ago
- Materials of the Nordic Probabilistic AI School 2022.☆176Updated 2 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 2 years ago
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆151Updated last year
- A Jax-based library for designing and training small transformers.☆286Updated 7 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆403Updated this week
- Materials of the Nordic Probabilistic AI School 2023.☆91Updated last year
- ☆246Updated 6 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆148Updated 6 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆375Updated this week
- Code for the book "The Elements of Differentiable Programming".☆79Updated last month
- ☆150Updated 8 months ago
- Parameter-Free Optimizers for Pytorch☆122Updated 11 months ago
- Annotated version of the Mamba paper☆481Updated last year
- Compositional Linear Algebra☆472Updated 2 weeks ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆254Updated this week
- Implementation of https://srush.github.io/annotated-s4☆489Updated 2 years ago
- ☆215Updated 9 months ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆168Updated last year
- ☆289Updated 3 months ago
- 🧱 Modula software package☆188Updated 3 weeks ago
- ☆166Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆173Updated this week
- Resources from the EleutherAI Math Reading Group☆53Updated last month
- Implementation of Diffusion Transformer (DiT) in JAX☆270Updated 10 months ago
- Recursive Bayesian Estimation (Sequential / Online Inference)☆58Updated last year