BordAX: A High-Performance JAX Framework for Programmatic Reinforcement Learning

Explore a JAX framework for running reinforcement learning experiments in environments defined programmatically.