pmap (1)
Date Rating Views Comments Length

Introduction to JAX (AI Adventures)

NumPy is fast, but how can we make it even faster?  In this article, we're going to look at a new library from Google Research called JAX and see how it can speed up machine learning. JAX can au...
16 May 2020 ·
1
· 156 · Alex
Introduction to JAX (AI Adventures)