Urgent.News

What's breaking now, across thousands of outlets.

AI

Async GRPO with LoRA across HF Jobs: a bucket, a proxy, and no NCCL

AsyncGRPOTrainer can now train a LoRA adapter and sync only that adapter to vLLM, eliminating the need for NCCL communication. This allows a rank-1 adapter for a 1.5B model, which is just a few megabytes in size, to be transferred via a storage bucket mounted in every HF Job instead of across machines. A small proxy in front of the vLLM replicas handles authentication, routing, and broadcast of adapter loads to each replica.

Five runs took the same recipe from 3 hours and 27 minutes to 53 minutes for 500 steps. LoRA support was recently added to TRL's AsyncGRPOTrainer, and it ships with TRL v1.14. Training and inference can now be run on separate machines, rather than sharing a machine.

Written by urgent.news from Hugging Face's reporting — not their text. Machine-written — may contain errors; check the original before relying on it.

Read the original at huggingface.co →

More in AI

More from Thursday 10 September →