[RFC]: Get rid of optim_bwd checks via wrapper.#2370

Open

krammnic wants to merge 2 commits intopytorch:mainfrom krammnic:optimizer_bwd

+126-64

Commits on Feb 9, 2025

add optimizer wrapper

Mark Obozov
authored and
Mark Obozov
committed

Commits on Feb 11, 2025

Update torchtune/training/_optimizer_wrapper.py

krammnic
and
joecummings
authored