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