Andrew Gelman personal blog

Defining Statistical Models in Jax?

This is Bob. And I’d like to know the best way for us to code a bunch of models in JAX to use to evaluate parallel algorithms including normalizing flows. I’m going to dump out my current thinking, but I’m really hoping to get feedback from experts on the best way to do this without starting a flame war in the…
Points
118
HN comments
18
Submissions
1
Cited by
0 commenters

Software EngineeringMath & Algorithms2024

On Hacker News

SubmittedTitle on HNPointsComments
Oct 13, 2024Defining Statistical Models in Jax? by hackandthink11818

More from Andrew Gelman

All 50 posts →

Related posts from the same era