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.

JAX is a Python library that Google researchers developed and introduced in 2018 for high-performance numerical computing.

JAX combines NumPy, automatic differentiation, and GPU/TPU support. In a new blog post, DeepMind researchers look at how JAX and its emergent ecosystem of open source libraries have served and accelerated an increasing number of machine learning projects.

tt ads