> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# GPT-NeoX / Pythia

[GPT-NeoX](https://github.com/EleutherAI/gpt-neox) is EleutherAI's large-scale language model architecture. The same `GPTNeoXForCausalLM` architecture is used by the Pythia scaling suite, OpenAssistant, Databricks Dolly, and StableLM models.

|                  |                                                 |
| ---------------- | ----------------------------------------------- |
| **Task**         | Text Generation                                 |
| **Architecture** | `GPTNeoXForCausalLM`                            |
| **Parameters**   | 1B – 20B                                        |
| **HF Org**       | [EleutherAI](https://huggingface.co/EleutherAI) |

## Available Models

* **GPT-NeoX-20B** (EleutherAI)
* **Pythia** suite: 70M, 160M, 410M, 1B, 1.4B, 2.8B, 6.9B, 12B (EleutherAI)
* **OA-SFT-Pythia-12B** (OpenAssistant)
* **Dolly-v2-12B** (Databricks)
* **StableLM-tuned-alpha-7B** (Stability AI)

## Architecture

* `GPTNeoXForCausalLM`

## Example HF Models

| Model                        | HF ID                                                                                                                     |
| ---------------------------- | ------------------------------------------------------------------------------------------------------------------------- |
| GPT-NeoX 20B                 | [`EleutherAI/gpt-neox-20b`](https://huggingface.co/EleutherAI/gpt-neox-20b)                                               |
| Pythia 12B                   | [`EleutherAI/pythia-12b`](https://huggingface.co/EleutherAI/pythia-12b)                                                   |
| OpenAssistant SFT Pythia 12B | [`OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5`](https://huggingface.co/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5) |
| Dolly v2 12B                 | [`databrickslabs/dolly`](https://github.com/databrickslabs/dolly)                                                         |
| StableLM tuned alpha 7B      | [`stabilityai/stablelm-tuned-alpha-7b`](https://huggingface.co/stabilityai/stablelm-tuned-alpha-7b)                       |

## Try with NeMo AutoModel

Install NeMo AutoModel and follow the fine-tuning guide to configure a recipe for this model.

**1. Install** ([full instructions](/get-started/installation)):

```bash
pip install nemo-automodel
```

**2. Clone the repo** to get example recipes you can adapt:

```bash
git clone https://github.com/NVIDIA-NeMo/Automodel.git
cd Automodel
```

**3. Fine-tune** by adapting a base LLM recipe — override the model ID on the CLI:

```bash
automodel --nproc-per-node=8 examples/llm_finetune/llama3_2/llama3_2_1b_squad.yaml \
  --model.pretrained_model_name_or_path <MODEL_HF_ID>
```

Replace `<MODEL_HF_ID>` with the model ID from **Example HF Models** above.

**1. Pull the container** and mount a checkpoint directory:

```bash
docker run --gpus all -it --rm \
  --shm-size=8g \
  -v $(pwd)/checkpoints:/opt/Automodel/checkpoints \
  nvcr.io/nvidia/nemo-automodel:26.06.00
```

**2.** The recipes are at `/opt/Automodel/examples/` — navigate there:

```bash
cd /opt/Automodel
```

**3. Fine-tune**:

```bash
automodel --nproc-per-node=8 examples/llm_finetune/llama3_2/llama3_2_1b_squad.yaml \
  --model.pretrained_model_name_or_path <MODEL_HF_ID>
```

See the [Installation Guide](/get-started/installation) and [LLM Fine-Tuning Guide](/recipes-e2e-examples/sft-peft).

## Fine-Tuning

See the [LLM Fine-Tuning Guide](/recipes-e2e-examples/sft-peft).

## Hugging Face Model Cards

* [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b)
* [EleutherAI/pythia-12b](https://huggingface.co/EleutherAI/pythia-12b)