Add value_and_grad option to all gradient-based solvers
Open
enhancement
- Dominant language
- Python
- Stars
- 1.1k
- Forks
- 76
- Avg merge
- 2d 21h
- Merged PRs (30d)
- 1
Description
This will allow the user to override the default behavior (jax.value_and_grad).
Contributor guide
Assessment
This issue has not been assessed yet.