Getting Started
Reference Documentation
Developer documentation
wax
API documentation
agent
Define API for Gym agents.
callbacks
Callbacks to work with wax.gym.gym_unroll
entity
Base class for Gym Agent and Env classes.
env
Define API for Gym environments.
haiku_agent
Gym agent defined from an Haiku module
haiku_env
Gym environment defined from an Haiku module