gpt_neo
This directory owns its Python builder, native runtime DSO, bundle section semantics, and tests. It has no source dependency on another family.
Task interfaces
Owned recipes
| Recipe | Checkpoint | Abstract task | Build | Manifest |
|---|---|---|---|---|
gpt-neo-125m | EleutherAI/gpt-neo-125m | text_generation | fp16; single device | families/gpt_neo/tests/manifests/gpt-neo-125m.json |
gpt-neo-125m-tp4 | EleutherAI/gpt-neo-125m | text_generation | fp32; TP4 | families/gpt_neo/tests/manifests/gpt-neo-125m-tp4.json |