fix adam perormance (#372)

Commit

5 years ago

fix adam perormance (#372) * fixing corner cases * revert to the previous perf for adam * adam high performance * run precommit