Skip to content

API reference

The API reference is auto-generated from the source code (docstrings and type hints) using mkdocstrings.

By area:

  • Models — GradiendModel, ParamMappedGradiendModel, ModelWithGradiend
  • Data — TextFilterConfig, TextPredictionDataCreator, TextPreprocessConfig
  • Training — TrainingArguments, Trainer, TextPredictionTrainer, Pruning configs
  • Evaluation — Evaluator, Encoder/Decoder, Visualizer

Use the navigation sidebar to open each area’s index and the listed classes.