GPT-J: Implement Memory Efficient Attention
Just a note that it would be great to add memory efficient attention ( https://github.com/AminRezaei0x443/memory-efficient-attention ) for larger models.  It might be nice to add an option to reuse the memory of input hidden states, too, if the user doesn't need them.
This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the [contributing guidelines](https://github.com/huggingface/transformers/blob/master/CONTRIBUTING.md) are likely to be ignored.