[[autodoc]] FlaxAlbertModel
    - call
FlaxAlbertForPreTraining
[[autodoc]] FlaxAlbertForPreTraining
    - call
FlaxAlbertForMaskedLM
[[autodoc]] FlaxAlbertForMaskedLM
    - call
FlaxAlbertForSequenceClassification
[[autodoc]] FlaxAlbertForSequenceClassification
    - call
FlaxAlbertForMultipleChoice
[[autodoc]] FlaxAlbertForMultipleChoice
    - call
FlaxAlbertForTokenClassification
[[autodoc]] FlaxAlbertForTokenClassification
    - call
FlaxAlbertForQuestionAnswering
[[autodoc]] FlaxAlbertForQuestionAnswering
    - call
