fix fastattn

update for tf2.4 (google#908)

* update for tf2.4

* fix mixed precision with recompute gradient

* update README

* fix multi gpus training

* update README

* fix LossScaleOptimizer bug

* disable steps_per_execution in default

* split all reduce

Fix ema (google#883)

* fix ema

* update README