elliotvilhelm/jax-policy-gradient

🤖 JAX implementation of REINFORCE policy gradient with baseline subtraction, entropy regularization, and gradient clipping for stable training.

2 best points · 0 comments · 1 Hacker News post · 4 GitHub stars · View on GitHub

Hacker News discussions

• Experimenting with policy gradient methods in Jax (2 points, 0 comments, by monadicmonad, 2025-07-08)