Skip to content
Change the repository type filter

All

    Repositories list

    • Lab Homepage
      CSS
      0000Updated Jan 28, 2026Jan 28, 2026
    • Official repo for "Predictability Enables Parallelization of Nonlinear State Space Models" by Gonzalez*, Kozachkov*, et al. Published NeurIPS '25.
      Python
      1200Updated Jan 16, 2026Jan 16, 2026
    • gpslds

      Public
      Project repo for gpSLDS
      Jupyter Notebook
      41700Updated Jan 12, 2026Jan 12, 2026
    • Tucker decompositions with normalization constraints
      Jupyter Notebook
      0100Updated Jan 12, 2026Jan 12, 2026
    • sing

      Public
      SING: SDE Inference via Natural Gradients
      Python
      43500Updated Dec 9, 2025Dec 9, 2025
    • Parallelizing MCMC across the sequence length via Newton's method.
      Python
      01800Updated Nov 25, 2025Nov 25, 2025
    • A PyTorch implementation of the PPSeq model with EM fitting and GPU acceleration.
      Jupyter Notebook
      2402Updated Oct 21, 2025Oct 21, 2025
    • elk

      Public
      Scalable and Stable Parallelization of Nonlinear RNNS
      Jupyter Notebook
      22800Updated Oct 21, 2025Oct 21, 2025
    • Official repo for "A Unifying Framework for Parallelizing Sequential Models with Linear Dynamical Systems"
      Python
      1500Updated Sep 25, 2025Sep 25, 2025
    • Code for the paper "Informed Correctors for Discrete Diffusion Models"
      Python
      0110Updated Sep 9, 2025Sep 9, 2025
    • ssm

      Public
      Bayesian learning and inference for state space models
      Jupyter Notebook
      225672311Updated May 9, 2025May 9, 2025
    • S5

      Public
      Python
      7131400Updated Jan 8, 2025Jan 8, 2025
    • Jupyter notebooks from our weekly (or so) hackathons
      Jupyter Notebook
      41100Updated Dec 3, 2024Dec 3, 2024
    • elk-torch

      Public
      Parallelizing nonlinear dynamics--in PyTorch
      Python
      1300Updated Dec 2, 2024Dec 2, 2024
    • elk-dev

      Public
      Scalable and Stable Parallelization of Nonlinear RNNS
      Python
      2200Updated Nov 26, 2024Nov 26, 2024
    • nm-rnn

      Public
      Work in progress...
      Jupyter Notebook
      1800Updated Oct 23, 2024Oct 23, 2024
    • forward

      Public
      Port Forwarding Utility (Linderman Lab edition)
      Shell
      29100Updated Oct 18, 2024Oct 18, 2024
    • salt

      Public
      Code accompanying "Switching Autoregressive Low-Rank Tensor Models", Lee et al., NeurIPS, 2023.
      Jupyter Notebook
      0000Updated Sep 30, 2024Sep 30, 2024
    • State Space Models library in JAX
      Jupyter Notebook
      105201Updated Sep 14, 2024Sep 14, 2024
    • fos

      Public
      Semi-NMF for Fos data
      Jupyter Notebook
      2100Updated Jul 9, 2024Jul 9, 2024
    • ssm-jax

      Public
      Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
      Jupyter Notebook
      662144Updated Jun 17, 2024Jun 17, 2024
    • Python
      21000Updated Jan 12, 2024Jan 12, 2024
    • Fast samplers for Neyman-Scott processes.
      Jupyter Notebook
      0911Updated Nov 28, 2023Nov 28, 2023
    • PPSeq.jl

      Public
      Neyman-Scott point process model to identify sequential firing patterns in high-dimensional spike trains
      Julia
      157001Updated Jul 18, 2023Jul 18, 2023
    • Whole-lifespan modelling of physical behavior
      Jupyter Notebook
      0140Updated Jun 25, 2023Jun 25, 2023
    • warhmm

      Public
      Python
      21200Updated Feb 27, 2023Feb 27, 2023
    • sixo

      Public
      Python
      0400Updated Nov 19, 2022Nov 19, 2022
    • ssm-docs

      Public
      Docs for SSM
      Jupyter Notebook
      0200Updated Oct 4, 2022Oct 4, 2022
    • jxf

      Public
      Exponential family distributions with TensorFlow probability and JAX
      Python
      0100Updated Apr 19, 2021Apr 19, 2021
    • logo

      Public
      Lab Logo
      Python
      1100Updated Nov 14, 2020Nov 14, 2020