[Feature] Implement AdamG optimizer

[kozistr]

Problem (Why?)

close #264

Solution (What/How?)

• implement AdamG optimizer
• optimize the memory usage of the TRAC optimizer

Other changes (bug fixes, small refactors)

N/A

Notes

N/A

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 100.00%. Comparing base (1054960) to head (d728f9e).
Report is 138 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff            @@
##              main      #265   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files           91        92    +1     
  Lines         6945      7007   +62     
=========================================
+ Hits          6945      7007   +62     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.