coax
Coax is a modular Reinforcement Learning (RL) Python package for solving OpenAI Gym (opens in new tab) environments with JAX (opens in new tab)-based function approximators (using Haiku (opens in new tab)).
coax: A Modular RL Package
Tools
Loading...