distributed reinforcement learning in JAX
Hi guys, Over the last couple of weeks and months I’ve been working on an RL library in Jax since most of the existing codebases didn’t really scale to my workflow. Now it’s in a kind of presentable state, so I’d love some feedback if someone wants to take a look. GitHub: https://github.com/noahfar…
Read the full story at r/reinforcementlearning ↗
Timeline · 2 reports
- 2026-10-11 22:13 · r/deeplearning
distributed reinforcement learning in JAX - 2026-10-11 22:11 · r/reinforcementlearning
distributed reinforcement learning in JAX