Kevin Wu

Flagpoles00

AI & ML interests

Always ready to learn!

Recent Activity

Organizations

MLX Community's profile picture

Flagpoles00's activity

reacted to Jaward's post with 👍 24 days ago
view post
Post
3867
Finally here it is: a faster, custom, scalable GRPO trainer for smaller models with < 500M params, can train on 8gb ram cpu, also supports gpu for sanity sake (includes support for vllm + flash attention). Using smolLM2-135M/360M-instructs as ref & base models. Experience your own “aha” moment 🐳 on 8gb ram.
Code: https://github.com/Jaykef/ai-algorithms/blob/main/smollm2_360M_135M_grpo_gsm8k.ipynb
  • 2 replies
·