I think the code snippets use *args and **kwargs just to demonstrate that you should pass all necessary arguments to the model and optimizer.
If that’s confusing, maybe a short note in the text saying that you should adapt the arguments to your use case would be nice. What do you think?