[TFAlbertForMaskedLM] is supported by this example script and notebook.