Skip to main content

Inference control plane for reasoning-aware open-source models

Project description

forge-infer

Metapackage bundling qwen-think and qwen3.6-mtp under a shared namespace.

pip install forge-infer pulls in qwen-think and qwen3.6-mtp as dependencies and re-exports their key APIs under a single forge namespace. This is packaging and narrative, not new code.

Why this exists

Two focused packages -- thinking-mode session control and MTP speculative decoding -- that belong together. forge-infer gives them a shared identity so you can recommend, install, and document them as a unit instead of scattering links across READMEs.

Install

pip install forge-infer

This installs both qwen-think and qwen3.6-mtp automatically.

Quick start

Thinking sessions (qwen-think)

Control when and how Qwen3.6 "thinks" -- budget tokens, toggle thinking on/off mid-conversation, route by complexity.

from forge.session import ThinkingSession

session = ThinkingSession(model="Qwen/Qwen3.6-27B")
response = session.chat("Explain merge sort", thinking=True)
print(response)

MTP speculative decoding (qwen3.6-mtp)

Tune multi-token prediction for throughput, find crossover points, generate backend configs.

from forge.mtp import recommend, quick_crossover, vllm_mtp_command, sglang_mtp_command
from forge.mtp import UseCase, Objective

# Get a recommendation for your hardware
rec = recommend(use_case=UseCase.SINGLE_USER, objective=Objective.MINIMIZE_LATENCY, gpu_id="rtx-4090")
print(rec.enable, rec.expected_gain)

# Find where MTP flips from positive to negative
for s in quick_crossover(gpu_id="rtx-3090"):
    print(f"MTP-{s.spec_tokens}: crossover at batch {s.crossover_batch_size}")

# Generate serve commands
print(vllm_mtp_command(model="Qwen/Qwen3.6-27B", num_speculative_tokens=2).command)
print(sglang_mtp_command(model="Qwen/Qwen3.6-27B", num_speculative_tokens=2).command)

Architecture

How the packages relate:

+---------------------------------------------+
|              forge (metapackage)             |
+------------------+--------------------------+
|   forge.session  |       forge.mtp          |
|  (qwen-think)    |   (qwen3.6-mtp)         |
|                  |                          |
|  Thinking-mode   |  MTP speculative decode  |
|  session control |  tuning & backend config |
+------------------+--------------------------+
|              Qwen3.6 model family           |
+---------------------------------------------+
  • forge.session -- Re-exports ThinkingSession from qwen-think.
  • forge.mtp -- Re-exports recommend, quick_crossover, vllm_mtp_command, sglang_mtp_command, UseCase, Objective from qwen3.6-mtp.

Individual packages

Package What it does
qwen-think Thinking-mode session management
qwen3.6-mtp MTP speculative decoding tuner

What this package does NOT do

  • No new functionality -- strictly re-exports from the underlying packages
  • No CLI -- the libraries are Python-first
  • No model generalization -- wraps Qwen3.6-specific versions as-is

License

Apache 2.0

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

forge_infer-0.3.0.tar.gz (21.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

forge_infer-0.3.0-py3-none-any.whl (10.1 kB view details)

Uploaded Python 3

File details

Details for the file forge_infer-0.3.0.tar.gz.

File metadata

  • Download URL: forge_infer-0.3.0.tar.gz
  • Upload date:
  • Size: 21.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for forge_infer-0.3.0.tar.gz
Algorithm Hash digest
SHA256 04ac6b1e6a87eb260f04d997a556218714c39aed96dc068e701d709f0f4b627b
MD5 ec91777baf42493b6422a4452c77e1e3
BLAKE2b-256 3dbcfdf192962b09cb418598b294c6d6fef779e906a31262b7d84e09073b0d47

See more details on using hashes here.

Provenance

The following attestation bundles were made for forge_infer-0.3.0.tar.gz:

Publisher: publish.yml on ArkaD171717/FORGE-Infer

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file forge_infer-0.3.0-py3-none-any.whl.

File metadata

  • Download URL: forge_infer-0.3.0-py3-none-any.whl
  • Upload date:
  • Size: 10.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for forge_infer-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 6dcf7c47f1cfef036a47cda1bf85ee8bf77aacb8bda327d49dea6dd07e0a216c
MD5 5ef3cacd7aa2b59bcc47bcd54b4009c3
BLAKE2b-256 98c56483edf091a21cd20f4957fe0f0e4f6abf7d6e6f29e9f66099b106b5155a

See more details on using hashes here.

Provenance

The following attestation bundles were made for forge_infer-0.3.0-py3-none-any.whl:

Publisher: publish.yml on ArkaD171717/FORGE-Infer

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page