More parameters

1. add ema
2. add AdamW, and more adjustable parameters in optimizer
3. add learning rate scheduler
4. add MLP radial
5. now radial mixing is applied on all (r, i, o) combinitions