Agents tagged "jax"

2 results
jax ×
Dev & Engineering

skrl: Modular Reinforcement Learning Library

A highly modular and readable RL library implemented in PyTorch, JAX, and NVIDIA Warp.

43 · Major gaps ★ 1.1k · today
Data & Analysis

Mava

A JAX research codebase for rapidly training and evaluating distributed multi-agent reinforcement learning systems.

41 · Major gaps ★ 938 · 1mo ago