Shypd.aiThe largest open index of AI toolsDynamax is an AI Agents & Automation tool that provides a Python package for probabilistic state space modeling using JAX. It supports inference and learning in various state space models, including Hidden Markov Models and Linear Dynamical Systems.
Trending
About
Dynamax is a Python package designed for probabilistic state space modeling, leveraging the JAX library for efficient computation. It offers robust capabilities for both inference (state estimation) and learning (parameter estimation) across a range of state space models. These include Hidden Markov Models (HMMs), Linear Gaussian State Space Models (Linear Dynamical Systems), Nonlinear Gaussian State Space Models, and Generalized Gaussian State Space Models with non-Gaussian emission models. The library provides both low-level, functionally pure inference algorithms and a user-friendly, object-oriented interface through model classes. It is compatible with other JAX ecosystem libraries like Optax for stochastic gradient descent and Blackjax for Hamiltonian Monte Carlo or sequential Monte Carlo.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
