[TFGPT2LMHeadModel] is supported by this causal language modeling example script and notebook.