Flax: Google’s Open Source Approach To Flexibility In Machine Learningby@fabian
2,836 reads

Flax: Google’s Open Source Approach To Flexibility In Machine Learning

March 26th 2020
6m
by @fabian 2,836 reads
tldt arrow
Read on Terminal Reader
Read this story w/o Javascript

Too Long; Didn't Read

Google researchers are developing Flax: Google’s Open Source Approach To Flexibility In Machine Learning. The base for the calculations is JAX instead of NumPy, which is also a Google research project. One of the biggest advantages of JAX is the use of XLA, a special. compiler for linear algebra, that enables execution on. GPUs and TPUs as well. Unfortunately you will still need Tensorflow at this point because Flax misses a good data input pipeline.

Companies Mentioned

Mention Thumbnail
Mention Thumbnail
featured image - Flax: Google’s Open Source Approach To Flexibility In Machine Learning
Fabian Deuser HackerNoon profile picture

@fabian

Fabian Deuser

does something with computer


Receive Stories from @fabian

react to story with heart

RELATED STORIES

L O A D I N G
. . . comments & more!