Image by Gerd Altmann from Pixabay

DeepMind just released Haiku and RLax for neural networks and reinforcement learning

Leverage the latest JAX libraries to facilitate your AI projects

Yong Cui
Towards Data Science
3 min readFeb 21, 2020

--

Artificial Intelligence (AI) research is evolving every single day, so are its tools — The cutting-edge AI company DeepMind just released two JAX-based libraries for neural networks (NN) and reinforcement learning…

--

--