Alex Lin Wang 王帅 — Research

Resources and advice for reinforcement learning written in JAX, including policy gradient architectures and jax.lax.scan speedups.