Bank
42 / 87
OptimizationMultiple choiceEasy

You wrote your own Adam with , and both moment estimates initialized to zero, and the update divides by directly.

The loss spikes in the first few hundred steps at a learning rate that is stable with the library's Adam.

What would you change in your optimizer?