    - call
[[autodoc]] FlaxTemperatureLogitsWarper
    - call
[[autodoc]] FlaxTopPLogitsWarper
    - call
[[autodoc]] FlaxTopKLogitsWarper
    - call
[[autodoc]] FlaxForcedBOSTokenLogitsProcessor
    - call
[[autodoc]] FlaxForcedEOSTokenLogitsProcessor
    - call
[[autodoc]] FlaxMinLengthLogitsProcessor
    - call
StoppingCriteria
A [StoppingCriteria] can be used to change when to stop generation (other than EOS token).
[[autodoc]] StoppingCriteria
    - call
[[autodoc]] StoppingCriteriaList
    - call
[[autodoc]] MaxLengthCriteria
    - call
[[autodoc]] MaxTimeCriteria
    - call
BeamSearch
[[autodoc]] BeamScorer
    - process
    - finalize
[[autodoc]] BeamSearchScorer
    - process
    - finalize
Utilities
[[autodoc]] top_k_top_p_filtering
[[autodoc]] tf_top_k_top_p_filtering
