Using lax.scan for time series in NumPyro

[vincentbt]

You can have a look at this code to add noise (parameter tau) at each timestep