Welcome to Baselax’s documentation!

Baselax (Baselines + jax) provides stable-baselines-style implementations of reinforcement learning (RL) algorithms with Google JAX framework.

Note

This project is under active development.

Contents

Tutorials

• Usage

APIs

• baselax.agents
  • baselax.agents.BaseAgent
  • baselax.agents.DQN
• baselax.utils
  • baselax.utils.network module
  • baselax.utils.seeding module