update `AdamW` optimizer. #333

Job	Run time
build (ubuntu-latest, 3.7, ==2.10)	35m 24s
build (ubuntu-latest, 3.7)	36m 35s
build (ubuntu-latest, 3.7, disable)	4m 59s
build (ubuntu-latest, 3.8, ==2.10)	37m 41s
build (ubuntu-latest, 3.8)	35m 1s
build (ubuntu-latest, 3.8, disable)	5m 17s
build (windows-latest, 3.7, ==2.10)	54m 28s
build (windows-latest, 3.7)	49m 19s
build (windows-latest, 3.7, disable)	6m 41s
build (windows-latest, 3.8, ==2.10)	52m 19s
build (windows-latest, 3.8)	50m 0s
build (windows-latest, 3.8, disable)	6m 59s
	6h 14m 43s

Provide feedback