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

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 (RL), namely Haiku and RLax, respectively.
Read at Medium
[
add
]
[
|
|
]