JAX

In a significant development for scientific computing, researchers have introduced JAXMg, a multi-GPU linear solver built within the JAX framework. This breakthrough allows the seamless integration of scalable linear algebra into JAX