JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning