Tag feed

#flax

2 posts0 followers

Building Neural Networks with Flax NNX

Feb 23 · 11 min read · Over the past two weeks, we've learned that JAX is fast (jit), that it eliminates loops (vmap), and that it computes gradients automatically (grad). These are powerful primitives. But if you've been f

Join discussion

WKWesley Kambalekambale.dev

0

TensorFlow v FLAX: A Comparison of Frameworks

Jun 12, 2023 · 10 min read · Introduction TensorFlow and FLAX are two popular frameworks that have gained significant traction in the deep learning community. In this article, we will explore and compare TensorFlow and FLAX, focusing on their features, functionality, advantages,...

Join discussion

#flax

Search Hashnode

#flax

Building Neural Networks with Flax NNX

TensorFlow v FLAX: A Comparison of Frameworks