Jarek Liesen

rejax. Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!

synthetic-gymnax. Drop-in environment replacements that make your RL algorithm train faster.

envelope. A jax-native environment interface

stochastic-hodgkin-huxley. Jupyter Notebook

Kinetix. Integrating stochasticity into reinforcement learning on general 2D physics environments in JAX.