Welcome to Jax Agents’s documentation!

Simple Jax-based Reinforcement Learning library.

For an introduction to jax-agents, start at the GitHub page.

Common

• Training
• Simulation and Rendering
• Plotting
• Data Processing and Replay Buffer
• Runge Kutta Integration
• Function Approximators

Algorithms

• Deep Deterministic Policy Gradient

Environments

• Pendulum

Indices and tables

• Index
• Module Index