How to store an optimizer in a struct field without type inference in Burn?

Looking at the Burn book about custom training loops. The Custom Type section has some examples. I also found the example code.