Skip to content

Commit b77a76e

Browse files
committed
fixing some formatting issues
1 parent a118945 commit b77a76e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paper/paper.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
---
2-
title: 'Dynamax: A Python package for probabilistic state space models (SSMs) written
3-
in JAX'
2+
title: 'Dynamax: A Python package for probabilistic state space models (SSMs) written in JAX'
43
tags:
54
- Python
65
- State space models
@@ -58,6 +57,7 @@ More information about state space models and algorithms for state inference and
5857
`Dynamax` is an open-source Python pacakge for state space modeling. Since it is built with `JAX` [@jax], it automatically supports just-in-time (JIT) compilation for hardware acceleration on CPU, GPU, and TPU machines. It also supports automatic differentiation for gradient-based model learning. While other libraries exist for state space modeling in Python (and some also use `JAX`), this library provides a combination of low-level inference algorithms and high-level modeling objects that can support a wide range of research applications.
5958

6059
The API for `Dynamax` is divided into two parts: a set of core, functionally pure, low-level inference algorithms, and a high-level, object oriented module for constructing and fitting probabilistic SSMs. The low-level inference API provides message passing algorithms for several common types of SSMs. For example, `Dynamax` provides `JAX` implementations for:
60+
6161
- Forward-Backward algorithms for discrete-state hidden Markov models (HMMs),
6262
- Kalman filtering and smoothing algorithms for linear Gaussian SSMs,
6363
- Extended and unscented Kalman filtering and smoothing for nonlinear Gaussian SSMs, and

0 commit comments

Comments
 (0)