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

What is a full example of using those structs to store the optimizer? I can’t find a way to do that. I either run into an error about setting the Record type, or it can’t match the type parameter because the init method returns an opaque type.