From a4fac292c2f02c8ef91df3de44a0c0d423419239 Mon Sep 17 00:00:00 2001 From: Scott Linderman Date: Sun, 13 Nov 2022 22:23:22 -0800 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index a679ed48..97d27d9d 100644 --- a/README.md +++ b/README.md @@ -114,7 +114,7 @@ for the case of an HMM with Gaussian emissions. (See [this notebook](https://github.com/probml/dynamax/blob/main/docs/notebooks/hmm/gaussian_hmm.ipynb) for a runnable version of this code.) -``` {.python} +```python import jax.numpy as jnp import jax.random as jr import matplotlib.pyplot as plt @@ -147,7 +147,7 @@ print(post.smoothed_probs.shape) # (1000, 3) JAX allows you to easily vectorize these operations with `vmap`. For example, you can sample and fit to a batch of emissions as shown below. -``` {.python} +```python from functools import partial from jax import vmap