pets-jax Jax implementation of Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models pip install -e .