Meta's Lucas Beyer and Sharad Vikram discuss the four-year-old implementation of JAX's stateful loop primitives · Digg