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.
submitted by /u/Yuqing7
[link] [comments]