DeepMind Augments, Salutes the JAX Library Ecosystem

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.

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.


#DataScientist, #DataEngineer, Blogger, Vlogger, Podcaster at . Back @Microsoft to help customers leverage #AI Opinions mine. #武當派 fan. I blog to help you become a better data scientist/ML engineer Opinions are mine. All mine.