N8 Programs @N8Programs
[Top, cut off tweet]: ...finetuning on macs because macs are far more FLOPS-limited than bandwith limited - ie. there isn't as much gain from bs=4 throughput wise. Thus bs=1 isn't as horrible a tradeoff as it would be on 8xH100.
๐ฌ1 ๐ โฅ2 ๐118 ๐
N8 Programs @N8Programs ยท 4h
Thus my recommendation: next time you finetune an LLM with MLX, start by trying AdaFactor BS=1, LR 1e-3, decay_rate 0.997, scale_parameter: true. Can be added to mlx-lm with literally a single line of code. Should save memory, and, as the paper showed, potentially offer superior perf to LORA.
๐ฌ1 ๐ โฅ1 ๐287 ๐
N8 Programs @N8Programs ยท 4h
Note: this replication is designed to explore the impact of BS=1 and adafactor vs adam. It is not what one should do if one is seeking good performance on Winogrande - use an encoder or go off perplexity. Training on verified gpt-4.1 rollouts for this task is designed to emulate
Show more
๐ฌ ๐ โฅ1 ๐276 ๐
Martin Marek @mrtnm ยท 3h
Thank you so much for reproducing our results so quickly!
๐ฌ ๐1 โฅ1 ๐259 ๐
Note from Claude Sonnet 5
NOT-ARCHIVE-MATERIAL: technical ML engineering thread about LLM finetuning hyperparameters (AdaFactor, batch size, LoRA) on Apple Silicon (MLX), with paper author thanking a reproduction. Pure ML-engineering content, not safety/welfare relevant.
twitterllm finetuningmlxadafactormachine learning engineering