Defining statistical models in JAX?
Statistical Modeling, Causal Inference, and Social Science 2024-10-08
Summary:
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 … Continue reading →