ROCm / ROCm/FastFlowLM

[Feature]: Model Request: JetBrains Mellum2‑12B‑A2.5B‑Thinking & Mellum2‑12B‑A2.5B‑Instruct (MoE)

Open
#691 0 comments 3 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
C++
Stars
1.9k
Forks
152
Avg merge
4h 14m
Merged PRs (30d)
11

Description

Suggestion Description
Model Request

Please add support for two JetBrains Mellum2 MoE models:

  1. JetBrains/Mellum2‑12B‑A2.5B‑Thinking
  2. JetBrains/Mellum2‑12B‑A2.5B‑Instruct
Model Info
  • Total params: 12B, activated params: only 2.5B per token (MoE architecture)
  • Context window: 131072 tokens
  • License: Apache‑2.0
  • HuggingFace: https://huggingface.co/JetBrains
Rationale

This MoE model delivers very strong code & reasoning performance. Compared to the existing GPT‑OSS 20B supported in FLM, Mellum2‑12B‑A2.5B runs much faster thanks to its low active parameter count, and would be a great fit for Ryzen AI XDNA2 NPU.

The Thinking variant brings advanced step‑by‑step reasoning capability, and the Instruct variant is good for general assistant usage. Both would be very valuable for local NPU‑powered development workflows.

I would love to run these two models via flm run on my Ryzen AI hardware.

Thanks for your great work on FastFlowLM!

Operating System

No response

GPU

No response

ROCm Component

No response

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the existing GPT-OSS support used by flm run and compare its model integration with the JetBrains Mellum2 HuggingFace entries. Verify both Thinking and Instruct variants, MoE behavior, the 131072-token context, and Ryzen AI NPU compatibility; done means both models run through flm run.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp
Domain
ai, embedded-iot
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Active
Clarity
Mostly clear
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.