JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
reinforcement-learning deep-learning deep-reinforcement-learning gym flax behavioral-cloning sac continuous-control deep-deterministic-policy-gradient jax soft-actor-critic offline-reinforcement-learning batch-reinforcement-learning
-
Updated
Oct 26, 2022 - Jupyter Notebook