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

Written by fabian | Published 2020/03/26
Tech Story Tags: machine-learning | mathematics | tensorflow | jax | flax | convolutional-neural-networks | deep-learning | hackernoon-top-story

TLDR 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.via the TL;DR App
no story
Written by fabian | does something with computer
Published by HackerNoon on 2020/03/26
ABOUTHow hackers start their afternoons. We publish tech stories and build publishing software.

READEntirely free library read by hundreds of millions to learn anything about any technology.

WRITEHackerNoon elevates quality technology writing far and wide across the interwebs.

PARTNERHackerNoon works directly with tech companies to place ads by content relevancy