Adafactor: Adaptive Learning Rates with Sublinear Memory Cost
Adam: A Method for Stochastic Optimization