Giters

fuyw / jrlzoo

A collection of RL baselines in Jax.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Jrlzoo

An RL codebase in JAX which focuses on efficiency and simplicity. Each baseline algorithm is self-contained in the corresponding folder.

On-policy RL

  • PPO

Off-policy RL

  • DDPG
  • DQN
  • SAC
  • TD3

Offline RL

  • COMBO
  • CQL
  • IQL
  • TD3BC

Fine-tuning Offline RL

  • AWAC
  • Off2OnRL
  • PEX

Meta RL

  • HypeRL
  • MAML
  • MQL
  • PEARL

Exploration

  • RND

Acknowledgements

About

A collection of RL baselines in Jax.

Languages

Language:Python 99.1%Language:Shell 0.6%Language:Ruby 0.4%