Search Packages

dynamax

A Python package for probabilistic state space modeling with JAX

mambular

DeepTab is a Python package that simplifies tabular deep learning by providing a suite of models for regression, classification, and distributional regression tasks. It includes models such as Mambular, TabM, FT-Transformer, TabulaRNN, TabTransformer, and tabular ResNets.

smcjax

Sequential Monte Carlo and particle filtering in JAX

deeptab

DeepTab is a Python package that simplifies tabular deep learning by providing a suite of models for regression, classification, and distributional regression tasks. It includes models such as Mambular, TabM, FT-Transformer, TabulaRNN, TabTransformer, and tabular ResNets.

rts-smoother

High-performance Rauch–Tung–Striebel Kalman smoother for the local level model

bayesnewton

Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.

discretax

State Space Models in Jax

linax

State Space Models in Jax

lrnnx

A unified PyTorch library for Linear RNNs