Using Deep Q-Learning (DQN) from Scratch Using RLax JAX Haiku and Optax to Train a CartPole Reinforcement Learning Agent
In this tutorial, we use a reinforcement learning agent RLaxa research-oriented library developed by Google DeepMind to build reinforcement learning algorithms with JAX. We combine RLax with JAX, Haiku, and Optax to build a Deep Q-Learning (DQN) agent that learns to solve the CartPole environment. Instead of using a fully packaged RL framework, we assemble … Read more