[EXTERNAL] Add vLLM Apriel2 model with plugin-based registration#447

Open

tscholak wants to merge 51 commits intomainfrom

feature/vllm-apriel2-models

Commits on Jan 10, 2026

Add Apriel2 conversion documentation and supernet pruning examples

tscholak
and
claude
committed

Commits on Jan 12, 2026

Add vLLM model implementation for Apriel2 with plugin-based registration

tscholak
and
claude
committed

Commits on Jan 15, 2026

Commits on Jan 16, 2026

Commits on Jan 17, 2026

Require CUDA kernels with no silent fallbacks

tscholak
and
claude
committed
Fix GDN chunk mode to use initial_state from cache

tscholak
and
claude
committed
Add extended cache tests for GDN and KDA equivalence

tscholak
and
claude
committed
Merge cache and non-cache equivalence tests with proper Apriel2Cache

tscholak
and
claude
committed
Refactor mixer tests and fix KDA mode selection

tscholak
and
claude
committed
Remove redundant test_causal_conv1d.py

tscholak
and
claude
committed
Consolidate cache.py into modeling_apriel2.py

tscholak
and
claude
committed
Enable bf16 tests and fix dtype handling

tscholak
and
claude
committed
Refactor test_mixer_equivalence.py: extract config fixtures and helpers

tscholak
and
claude
committed

Commits on Jan 18, 2026

Merge branch 'fix/require-cuda-kernels-no-fallbacks' into feature/vllm-apriel2-models
tscholak
committed

Commits on Jan 20, 2026

Commits on Jan 21, 2026

Commits on Jan 22, 2026

Commits on Jan 23, 2026

Commits on Jan 26, 2026

Add dynamic layer placement support for stochastic mixers

tscholak
and
claude
committed

Commits on Jan 27, 2026

Merge origin/main into feature/vllm-apriel2-models
tscholak
committed

Commits on Jan 28, 2026

Commits on Feb 5, 2026

Cache reset between layer placements (#468)
oleksost
authored

Commits on Feb 17, 2026

fix for loading supernet with vllm and default mixer gdn
oleksost
committed

Commits on Feb 20, 2026

remap weights when loading from supernet checkpoint with predefined placement
oleksost
committed

Commits on Feb 24, 2026

Thin-down supernets
nandahkrishna
committed