3 projects
jumanji
A diverse suite of scalable reinforcement learning environments in JAX
flashbax
Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.
id-marl-eval
A Python library for Multi-Agent Reinforcement Learning evaluation.
