Add AdamW to the optimization search space

[cooperlab]

This optimizer provides exponential model averaging over multiple epochs and is a better alternative than tfa which will be discontinued.

[create-issue-branch[bot]]

Branch issue-21-Add_AdamW_to_the_optimization_search_space created!