Brax v0.14.2 Release Notes
- Distributional PPO Support: Added a distributional critic to capture return
variance, leading to more reliable advantage estimates and improved policy
updates (reference: https://arxiv.org/abs/1710.10044). - Parametrize CNN networks in brax PPO.