Module @mlx-bun/training

Interfaces

TrainConfig
TrainResult

Variables

DEFAULT_TRAIN_CONFIG

Functions

trainLora

References

AdamW → AdamW
AdamWOptions → AdamWOptions
attachForTraining → attachForTraining
buildTrainableLora → buildTrainableLora
DatasetFormat → DatasetFormat
DEFAULT_TARGET_MODULES → DEFAULT_TARGET_MODULES
detachTraining → detachTraining
DiffusionLoraConfig → DiffusionLoraConfig
diffusionLoss → diffusionLoss
DiffusionPair → DiffusionPair
disposeLora → disposeLora
DpoBatch → DpoBatch
dpoChosenLength → dpoChosenLength
DpoExample → DpoExample
dpoLoss → dpoLoss
dpoMetrics → dpoMetrics
dpoRefLogps → dpoRefLogps
dpoRejectedLength → dpoRejectedLength
encodeDpoRow → encodeDpoRow
encodeSftRow → encodeSftRow
exportAdapter → exportAdapter
ExportManifest → ExportManifest
flatA → flatA
flatB → flatB
flatParams → flatParams
fuseAdapter → fuseAdapter
FuseOptions → FuseOptions
FuseProgress → FuseProgress
FuseStats → FuseStats
iterateDpoBatches → iterateDpoBatches
iterateSftBatches → iterateSftBatches
loadDpoDataset → loadDpoDataset
loadSftDataset → loadSftDataset
mergeAdapters → mergeAdapters
MergeStats → MergeStats
orpoLoss → orpoLoss
orpoMetrics → orpoMetrics
PAD_TO → PAD_TO
probeFormat → probeFormat
RankScaling → RankScaling
resolveRanks → resolveRanks
ResolveRanksOptions → ResolveRanksOptions
rowLength → rowLength
saveAdapter → saveAdapter
SaveAdapterConfig → SaveAdapterConfig
setLoraScale → setLoraScale
SftBatch → SftBatch
SftExample → SftExample
sftLoss → sftLoss
sftRowFormat → sftRowFormat
SftRowFormat → SftRowFormat
SftScope → SftScope
tokenizePair → tokenizePair
TrainableLora → TrainableLora
TrainableTarget → TrainableTarget
trainDiffusionLora → trainDiffusionLora
TrainingProgress → TrainingProgress
TrainingProgressCallback → TrainingProgressCallback
warmupCosineSchedule → warmupCosineSchedule