Přeskočit na obsah
ejaj.cz

train-llm-from-scratch

ZdarmaOpen sourceVývojářské nástroje

Open-source dokumentace a kód pro trénování jazykového modelu (LLM) od nuly v čistém PyTorchi, včetně post-tréninku a zarovnání.

fareedkhan-dev.github.io
Náhled nástroje train-llm-from-scratch

O nástroji

Projekt krok za krokem provádí celým pipeline tréninku LLM: pretraining na datasetu The Pile (9,8 miliardy tokenů, model o cca 400M parametrech), následné SFT na datasetech Alpaca, Dolly a GSM8K, trénink reward modelu a zarovnání pomocí DPO/ORPO/KTO, PPO nebo GRPO. Veškerý kód je napsán ručně v PyTorchi bez použití knihoven trl, peft nebo transformers a lze jej spustit na jednom GPU nebo škálovat přes více GPU pomocí DDP. Dokumentace je rozdělena do sekcí Foundations (tokenizace, transformer, attention, objektivy, optimalizace, generování) a jednotlivých stránek pro každou fázi pipeline (data, pretraining, SFT, reward model, DPO, PPO, GRPO, evaluace, inference/chat) a je v angličtině. Celý řetězec lze spustit jedním skriptem run_posttraining.sh, který provede SFT → RM → DPO → PPO → GRPO → evaluaci.

Klíčové vlastnosti

  • Ruční PyTorch implementace bez trl, peft či transformers
  • Pretraining na 9,8 miliardy tokenů z The Pile
  • SFT na datasetech Alpaca, Dolly a GSM8K
  • Zarovnání pomocí DPO, ORPO, KTO, PPO nebo GRPO
  • Trénink reward modelu metodou Bradley-Terry
  • Spustitelné na jednom GPU nebo škálované přes DDP

Časté dotazy

Co přesně tento projekt umí?

Provede vás celým procesem trénování LLM od nuly v čistém PyTorchi – od pretrainingu na datasetu The Pile přes SFT až po zarovnání pomocí DPO, PPO nebo GRPO, včetně evaluace a inference.

Kolik to stojí?

Jde o open-source dokumentaci a kód zdarma ke stažení a použití, náklady vznikají pouze za vlastní výpočetní výkon (GPU) potřebný k trénování.

Pro koho je to určené a jak začít?

Je určeno lidem, kteří chtějí porozumět vnitřnímu fungování trénování LLM; začít se doporučuje sekcí Foundations (tokenizace, transformer, attention) a poté projít pipeline pomocí skriptu run_posttraining.sh.

Kategorie

AI novinky do e-mailu

Každý pátek 3 vybrané AI nástroje, prompt týdne a to nejdůležitější ze světa AI – přehledně, česky a bez balastu.

Jeden z nejstarších pravidelných českých AI newsletterů.

Jednou týdně, v pátek. Žádný spam – odhlásíte se jedním klikem.

Podobné nástroje

Hledáte alternativu k train-llm-from-scratch? Tady je 6 podobných nástrojů z kategorie Vývojářské nástroje.

Zobrazit všechny alternativy v kategorii Vývojářské nástroje

Další nástroje v kategorii Vývojářské nástroje

Nezůstávejte jen u nejoblíbenějších – tohle jsou další nástroje z kategorie Vývojářské nástroje, které stojí za prohlédnutí.