Documentation on how to write training code?

[BrechtCorbeel]

I'm at the end of models I find interesting, but I think I might want to train with AMD is there any documentation on how train flux that might leave me a clue on how to implement rocM for training?

I still likely will use NVIDIA so just documentation would be great to see what I can do with the model.