Skip to content

Migrate boltz2 to the semantic Task SDK #1434

Open

Description

@yifeif-nv

Parent epic: #1430

Status: Not started — comment here before beginning work.

Family: families/boltz2/
Current declared task mode(s): structure_prediction
Target semantic Task(s): Identify the existing semantic Task contract(s) that preserve every current capability; record the exact IDs in the PR.

Work

  • Inventory this family's primary and secondary behavior, including batch, streaming, raw outputs and stateful operations where supported. Record the current defaults, profiles, checkpoint cases, and reference criteria.
  • Implement and bind the matching semantic Task interface(s) in this family's runtime. Preserve its graph, preprocessing, inference, postprocessing, configuration and complete owned results.
  • Align family-owned support, build metadata and manifests with the primary Task. Advertise only interfaces implemented by the loaded bundle; document whether existing bundles need rebuilding.
  • Add family-owned CPU contracts and public C/C++ SDK consumers. Cover invalid inputs, configuration handling, result lifetime and every relevant lifecycle.
  • Run a selected real checkpoint E2E with the existing oracle and thresholds, then check the remaining declared capabilities and public CLI, example and benchmark paths. Preserve meaningful passing criteria.

PR evidence

Link the migration PR here. Report the tested revision, exact commands and results, selected checkpoint cases, original reference checks, current-head CI, and any untested configurations. Keep a normal migration within families/boltz2/**; raise a separate linked prerequisite issue if a shared contract is genuinely missing. Do not weaken an acceptance criterion to pass CI.

References: #1226 · family validation guide.

Activity

  1. roma5087 commented on Sep 29, 2026

    @roma5087

    Hi, I'd like to take the boltz2 migration.

    families/boltz2/ on current main already has Boltz2Pipeline implementing
    IStructurePrediction, and plugin.cpp, support.py and model.py all use
    structure_prediction. So I plan to do the following, modeled on the merged
    migration in #1308:

    • Add the family's bind record / IModel binding, if it isn't already
      covered. I'll confirm this against the current runtime first.
    • Add family-owned CPU tests for the Task contract: binding, request
      validation, complete result ownership (samples, confidence, metadata) and
      unsupported config.
    • Add public C11 and C++17 SDK consumers to the existing E2E. Both will read
      the result after releasing the model and must agree on the structure output
      and confidence values.
    • Align the manifest and support metadata with the Task, and add a family
      README.
    • Run the existing real-checkpoint E2E on a single GPU with the current oracle
      and thresholds. I'll report the exact revision, commands, results, and
      anything I couldn't test.

    I'll keep all changes inside families/boltz2/**. I'll open a separate
    prerequisite issue if I find a shared contract is missing, and I won't weaken
    any existing acceptance criterion. I'll link a single migration PR here.

  2. yifeif-nv commented on Sep 30, 2026

    @yifeif-nv
    CollaboratorAuthor

    Hi, I'd like to take the boltz2 migration.

    families/boltz2/ on current main already has Boltz2Pipeline implementing IStructurePrediction, and plugin.cpp, support.py and model.py all use structure_prediction. So I plan to do the following, modeled on the merged migration in #1308:

    • Add the family's bind record / IModel binding, if it isn't already
      covered. I'll confirm this against the current runtime first.
    • Add family-owned CPU tests for the Task contract: binding, request
      validation, complete result ownership (samples, confidence, metadata) and
      unsupported config.
    • Add public C11 and C++17 SDK consumers to the existing E2E. Both will read
      the result after releasing the model and must agree on the structure output
      and confidence values.
    • Align the manifest and support metadata with the Task, and add a family
      README.
    • Run the existing real-checkpoint E2E on a single GPU with the current oracle
      and thresholds. I'll report the exact revision, commands, results, and
      anything I couldn't test.

    I'll keep all changes inside families/boltz2/**. I'll open a separate prerequisite issue if I find a shared contract is missing, and I won't weaken any existing acceptance criterion. I'll link a single migration PR here.

    Sounds good. Please go ahead and start a PR and link it to this issue!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions