4 ms·
I wonder if this change has any implications for the TensorFlow vs. JAX situation/transition. IIRC I read that DeepMind mainly used JAX, but not sure about the
by agnosticmantis 3y ago
I wonder if this change has any implications for the TensorFlow vs. JAX situation/transition.
IIRC I read that DeepMind mainly used JAX, but not sure about the Brain.
Any insights from the people in the know? It seems JAX is the future, but TF dominates current production stacks.
- arkano 3y agoFlax, a big neural network JAX library, is developed and used by Brain. TF will stay around for boilerplate / data-loading, but JAX support probably isn’t going anywhere.