[optim] NAdam fold state updates into functional (#71334)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71334
Test Plan: Imported from OSS
Reviewed By: anjali411
Differential Revision: D33767864
Pulled By: mikaylagawarecki
fbshipit-source-id: 4d985e9e346f40110bd4231e0f16e5643fbc448d
(cherry picked from commit 58aa77e367f7c863563c0469ef9df818236ed57c)