It’s a JAX, JAX, JAX, JAX World

This post is by Bob. The title is based on the similarly named classic film.   “Big” models moving from Stan to JAX Ever since the big ML frameworks PyTorch and TensorFlow were released, the Stan developers have been worried … Continue reading → …