AXLEARN

Summary from the TLDR AI newsletter

A library built on top of JAX and XLA that supports the development of large-scale deep learning models.