checkpointed while loops

TL;DR: Main AI News: Equinox is a cutting-edge JAX library that has been making waves in the data science and...