[FEATURE] Access to logits and final hidden layer

[lshamis]

Hello, I don't believe there is any way to access logits or the final hidden layer embedding in the current API and would like to request this feature. Thank you