add type annotations for `gptj`
# What does this PR do?

This PR adds type annotations for GPT-J (PyTorch) as described in #16059.

## Who can review?

@Rocketknight1

Anyone in the community is free to review the PR once the tests have passed.


_The documentation is not available anymore as the PR was closed or merged._