All modules for which code is available
- functools
- torchtune.config._instantiate
- torchtune.config._parse
- torchtune.config._utils
- torchtune.config._validate
- torchtune.data._chat_formats
- torchtune.data._converters
- torchtune.data._instruct_templates
- torchtune.data._types
- torchtune.data._utils
- torchtune.datasets._alpaca
- torchtune.datasets._chat
- torchtune.datasets._cnn_dailymail
- torchtune.datasets._concat
- torchtune.datasets._grammar
- torchtune.datasets._instruct
- torchtune.datasets._packed
- torchtune.datasets._preference
- torchtune.datasets._samsum
- torchtune.datasets._slimorca
- torchtune.datasets._stack_exchanged_paired
- torchtune.datasets._text_completion
- torchtune.datasets._wikitext
- torchtune.models.code_llama2._model_builders
- torchtune.models.gemma._component_builders
- torchtune.models.gemma._model_builders
- torchtune.models.gemma._tokenizer
- torchtune.models.llama2._component_builders
- torchtune.models.llama2._model_builders
- torchtune.models.llama2._tokenizer
- torchtune.models.llama3._component_builders
- torchtune.models.llama3._model_builders
- torchtune.models.llama3._tokenizer
- torchtune.models.llama3_1._component_builders
- torchtune.models.llama3_1._model_builders
- torchtune.models.mistral._component_builders
- torchtune.models.mistral._model_builders
- torchtune.models.mistral._tokenizer
- torchtune.models.phi3._component_builders
- torchtune.models.phi3._model_builders
- torchtune.models.phi3._tokenizer
- torchtune.modules.attention
- torchtune.modules.common_utils
- torchtune.modules.feed_forward
- torchtune.modules.kv_cache
- torchtune.modules.loss.dpo
- 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.tokenizers._utils
- torchtune.modules.transformer
- torchtune.utils._checkpointing._checkpointer
- torchtune.utils._checkpointing._checkpointer_utils
- torchtune.utils._device
- torchtune.utils._distributed
- torchtune.utils._generation
- torchtune.utils._profiler
- torchtune.utils._version
- torchtune.utils.argparse
- torchtune.utils.collate
- torchtune.utils.logging
- torchtune.utils.memory
- torchtune.utils.metric_logging
- torchtune.utils.precision
- torchtune.utils.quantization
- torchtune.utils.seed
- typing