Shypd.aiThe largest open index of AI toolsjaxrl is an open-source AI Agents & Automation tool that provides JAX (Flax) implementations of deep reinforcement learning algorithms. It focuses on continuous action spaces and includes algorithms like Soft Actor Critic and Advantage Weighted Actor Critic.
Trending
About
jaxrl is an open-source repository offering JAX (Flax) implementations of various deep reinforcement learning algorithms, specifically designed for continuous action spaces. The project aims to provide simple and clean implementations for research and development, rather than for baseline results. Key algorithms included are Soft Actor Critic with learnable temperature, Advantage Weighted Actor Critic, Image Augmentation Is All You Need (for K=1, M=1), Deep Deterministic Policy Gradient with Clipped Double Q-Learning, Randomized Ensembled Double Q-Learning, and Behavioral Cloning. The repository emphasizes ease of use for building upon existing research and provides detailed installation instructions, including prerequisites for Python, Poetry, and MuJoCo, as well as guidance for GPU support and Docker deployment.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
