Our paper on JAXMg, a distributed multi-GPU and multi-node linear algebra library for JAX, is now on arXiv.