Skip to content

API Reference

Auto-generated from source docstrings.

Package Description
Core ConfigurableBlock, registries, attention, FFN, norm, position
Models LanguageModel, config factories, generation
Training Trainer, optimizers, checkpoints, callbacks, DPO, GRPO, MTP
Data Tokenizers, datasets, batching
Eval Perplexity, bits-per-byte metrics
Inference Sampling, speculative decoding, beam search, reward model