[Fix] should_grokfast condition

[kozistr]

Problem (Why?)

Fixes #279
Close #194, #278

Solution (What/How?)

• only check grokfast, grokfast_lamb params when initializing state.

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 (09a2b83) to head (600899f).
Report is 6 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff            @@
##              main      #280   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files           94        94           
  Lines         7237      7237           
=========================================
  Hits          7237      7237           

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