Implementing a Transformer Encoder from Scratch with JAX and Haiku 🤖
Tags
3 pages
Implementation
Vectorize and Parallelize RL Environments with JAX, Q-learning at the Speed of Light⚡
Temporal-Difference Learning and the importance of exploration, an illustrated guide
