# gpt_neox

| | |
|---|---|
| **Model type** | `gpt_neox` |
| **Class** | `GPTNeoXCausalLMModel` |
| **Task** | `text-generation` |
| **Source** | `models/gpt_neox.py` |

## Description

GPT-NeoX / Pythia causal language model.

## Usage

```bash
mobius build --model <MODEL_ID> output_dir/
```

```python
from mobius import build

model = build("<MODEL_ID>")
```