Posts

Rainbow, The Colorful Evolution of Deep Q-Networks
Everything you need to assemble the DQN Megazord in JAX.
Last updated on Jul 13, 2024 1 min read
Rainbow, The Colorful Evolution of Deep Q-Networks
Breaking down State-of-the-Art PPO Implementations in JAX
All the tricks and details you wish you knew about Proximal Policy Optimization
Last updated on May 14, 2024 1 min read
Breaking down State-of-the-Art PPO Implementations in JAX
A Gentle Introduction to Deep Reinforcement Learning in JAX
Solving the CartPole environment with DQN in under a second
Last updated on May 14, 2024 1 min read
A Gentle Introduction to Deep Reinforcement Learning in JAX
Implementing a Transformer Encoder from Scratch with JAX and Haiku
Understanding the fundamental building blocks of Transformers.
Last updated on May 14, 2024 1 min read
Implementing a Transformer Encoder from Scratch with JAX and Haiku
Vectorize and Parallelize RL Environments with JAX, Q-learning at the Speed of Light
Solving the CartPole environment with DQN in under a second
Last updated on May 14, 2024 1 min read
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
Comparing model-free and model-based RL methods on a dynamic grid world
Last updated on May 14, 2024 1 min read
Temporal-Difference Learning and the importance of exploration, An illustrated guide