All modules for which code is available
- functools
- torchtune.config._instantiate
- torchtune.config._parse
- torchtune.config._validate
- torchtune.data._chat_formats
- torchtune.data._instruct_templates
- torchtune.data._utils
- torchtune.datasets._alpaca
- torchtune.datasets._grammar
- torchtune.datasets._samsum
- torchtune.datasets._slimorca
- torchtune.models.gemma._model_builders
- torchtune.models.llama2._model_builders
- torchtune.models.llama3._model_builders
- torchtune.models.mistral._model_builders
- torchtune.modules.attention
- torchtune.modules.common_utils
- 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.tokenizers._sentencepiece
- torchtune.modules.tokenizers._tiktoken
- torchtune.modules.transformer
- torchtune.utils._checkpointing._checkpointer
- torchtune.utils._device
- torchtune.utils._distributed
- torchtune.utils._profiler
- torchtune.utils.argparse
- torchtune.utils.collate
- torchtune.utils.logging
- torchtune.utils.memory
- torchtune.utils.metric_logging
- torchtune.utils.precision
- torchtune.utils.seed