Skip to content

Commit

Permalink
Documented gradient noise feature in jax planners
Browse files Browse the repository at this point in the history
  • Loading branch information
mike-gimelfarb committed Oct 30, 2024
1 parent 5169235 commit 35ad639
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions docs/jax.rst
Original file line number Diff line number Diff line change
Expand Up @@ -339,6 +339,10 @@ The full list of settings that can be specified in the configuration files are a
- Batch size for training
* - clip_grad
- Clip gradients to within a given magnitude
* - noise_grad_eta
- Scale of the gradient noise variance
* - noise_grad_gamma
- Decay rate of the gradient noise variance
* - compile_non_fluent_exact
- Model relaxations are not applied to non-fluent expressions
* - cpfs_without_grad
Expand Down

0 comments on commit 35ad639

Please sign in to comment.