[N] DeepMind Augments, Salutes the JAX Library Ecosystem

Google’s UK-based lab and research company DeepMind has added Jraph to the growing number of open-sourced libraries around JAX, while surveying the machine learning framework’s development and ecosystem.

Here is a quick read: DeepMind Augments, Salutes the JAX Library Ecosystem

DeepMind is continuously updating the JAX ecosystem to accelerate ML studies. The open-sourced libraries can be found at the project GitHub.