Jax

SBX: Stable Baselines Jax

Proof of concept version of Stable-Baselines3 in Jax.