Download Latest Version v0.14.0 source code.tar.gz (124.1 MB)
Email in envelope

Get an email when there's a new version of Brax

Home / v0.12.3
Name Modified Size InfoDownloads / Week
Parent folder
README.md 2025-04-11 510 Bytes
v0.12.3 source code.tar.gz 2025-04-11 124.3 MB
v0.12.3 source code.zip 2025-04-11 124.6 MB
Totals: 3 Items   248.9 MB 0

Brax v0.12.3 Release Notes

  • Add training metrics to brax PPO, which allows users to avoid running evals during training while getting more frequent metric updates (à là RSL-RL). Set num_evals=0 and log_training_metrics=True.
  • Add checkpointing directly to brax PPO, rather than relying on the policy_params_fn callback.
  • Fix bug in inverted pendulum (#574) where the position of the tip was being mis-calculated.
  • Add UInt64 to prevent overflow of training steps in brax training. Fixes [#578].
Source: README.md, updated 2025-04-11