[CB] Add an option to return logprobs (#44835)
* Pretty much the feature
* Cleanup
* Add custom flag
* Add test
* Rebase changes
* Review compliance
* Review compliance 2
* Style
* more style
* Relaxed cuda graphs test
* Style and fix default attn
* docstirng
* style