Quickstart in readme.md should include pattern of how to do inference?

[cottrell]

Unless I am completely missing reading something, it seems like the initial quickstart example should have a line showing how to do inference from the model that is defined inside the loss function.

I can imagine different ways of doing this but it feels like the quickstart should be a bit opinionated about the standard way to accomplish this.

[IanQS]

I think it does show how one can do it? Check out this colab link that I made.

Basically, forward.apply does the inference for you

[cottrell]

It is further below, I just meant the quickstart section should have a forward function perhaps.

Maybe I should have just edited the README and added a PR for discssion.