All modules for which code is available
- torchtune.config._instantiate
- torchtune.config._parse
- torchtune.datasets._alpaca
- torchtune.datasets._grammar
- torchtune.datasets._samsum
- torchtune.datasets._slimorca
- torchtune.models.llama2._component_builders
- torchtune.models.llama2._model_builders
- torchtune.modules.attention
- torchtune.modules.feed_forward
- torchtune.modules.kv_cache
- torchtune.modules.lr_schedulers
- torchtune.modules.peft.lora
- torchtune.modules.peft.peft_utils
- torchtune.modules.position_embeddings
- torchtune.modules.rms_norm
- torchtune.modules.tokenizer
- torchtune.modules.transformer
- torchtune.utils._checkpointing._checkpointer
- torchtune.utils._device
- torchtune.utils._distributed
- torchtune.utils.argparse
- torchtune.utils.collate
- torchtune.utils.logging
- torchtune.utils.memory
- torchtune.utils.metric_logging
- torchtune.utils.precision
- torchtune.utils.seed