Migrate EMA backend to lucidrains' ema-pytorch

[dmarx]

https://github.com/lucidrains/ema-pytorch/blob/main/ema_pytorch/ema_pytorch.py

Current EMAImage assumes we only need to track EMA on a single tensor, which would preclude EMA-ing a DIP image model (or any other implicit model.