model-free-rl
Massively Parallel Deep Reinforcement Learning. 🔥
Deep reinforcement learning algorithms and environments, written completely in JAX + Flax NNX for end-to-end GPU-accelerated training.
