# tiny-RemoteForCausalLM

Hugging Face: `trl-internal-testing/tiny-RemoteForCausalLM`
Reconstructed from `config.json` alone. No weights were read.

| | |
|---|---|
| Derived from structure | 2,182,312 |
| In the published checkpoint | 2,054,056 (safetensors.total, 2026-09-08) |
| Delta | +6.24% |
| Comparability | custom-code |
| Layers | 16 |
| Priced on | A10G (24GB) |
| Est. one run | $15.37 |

> This repository ships its own modeling code (`auto_map`, e.g. `configuration_remote.py`), so `config.json` names a class in the repo rather than an architecture `transformers` defines. The graph below is what those config keys mean under `transformers` semantics, which is not necessarily what the repo's own file builds. A gap here is a statement about what we read, not about the checkpoint.

## Structure

| # | Layer | Type | Output shape |
|---|---|---|---|
| 1 | Input | Input | 1 × 131072 |
| 2 | Embedding | Embedding | 1 × 131072 × 8 |
| 3 | RoPE | RoPE | 1 × 131072 × 8 |
| 4 | RMSNorm_1_1 | RMSNorm | 1 × 131072 × 8 |
| 5 | Attention_1 | Grouped Query Attn | 1 × 131072 × 8 |
| 6 | Add_1_attn | Add | 1 × 131072 × 8 |
| 7 | RMSNorm_1_2 | RMSNorm | 1 × 131072 × 8 |
| 8 | FFN_1 | SwiGLU | 1 × 131072 × 8 |
| 9 | Add_1_ffn | Add | 1 × 131072 × 8 |
| 10 | RMSNorm_2_1 | RMSNorm | 1 × 131072 × 8 |
| 11 | Attention_2 | Grouped Query Attn | 1 × 131072 × 8 |
| 12 | Add_2_attn | Add | 1 × 131072 × 8 |
| 13 | RMSNorm_2_2 | RMSNorm | 1 × 131072 × 8 |
| 14 | FFN_2 | SwiGLU | 1 × 131072 × 8 |
| 15 | Add_2_ffn | Add | 1 × 131072 × 8 |
| 16 | Final_RMSNorm | RMSNorm | 1 × 131072 × 8 |
| 17 | LM_Head | Linear | 1 × 131072 × 128256 |
| 18 | Output | Output | 1 × 131072 × 128256 |

## Do this to your own model

```
pip install neurarch-trace
neurarch-trace trl-internal-testing/tiny-RemoteForCausalLM --plan --share
```

- Page: https://neurarch.com/m/tiny-remoteforcausallm.html
- All models: https://neurarch.com/m/index.json
- The checks: https://neurarch.com/r/index.json