Categories
2 pages
JAX
Implementing a Transformer Encoder from Scratch with JAX and Haiku 🤖
Vectorize and Parallelize RL Environments with JAX, Q-learning at the Speed of Light⚡
