torchopt
torchopt copied to clipboard
[Feature Request] AdamW support
Motivation
for training big model
Solution
Checklist
- [x] I have checked that there is no similar issue in the repo (required)