←back to thread

Defining Statistical Models in Jax?

(statmodeling.stat.columbia.edu)
106 points hackandthink | 1 comments | 13 Oct 24 15:20 UTC | HN request time: 0.203s | source
1. techwizrd ◴[17 Oct 24 14:14 UTC] No.41869882[source]▶
This is coming at the perfect time! I was recently trying to decide whether I wanted to implement a model in Stan or Pyro/Numpyro, and I've been eyeing implementing in JAX. I would love to write a tutorial comparing Stan to Jax.