Address convergence and nan issues

[jundi69]

Does the following:

1. Gradient clip after allreduce to enhance stability of gradients
2. Zero outer optimizer grads to avoid accumulating gradients in the outer optimizer
3. Remove gradscaler as we are training in bfloat16
4. Reduce effective batch size back to 512 as per diloco paper