flax

70571d21 - Move JAX-RL community example from deprecated section to Linen section