Skip to content

Sync dev updates: SafeTensors model B support, legacy color-code circuit, docs and CI fixes#90

Merged
ivanbasov merged 1 commit into
mainfrom
sync/dev-updates-2026-07-14
Jul 15, 2026
Merged

Sync dev updates: SafeTensors model B support, legacy color-code circuit, docs and CI fixes#90
ivanbasov merged 1 commit into
mainfrom
sync/dev-updates-2026-07-14

Conversation

@ivanbasov

@ivanbasov ivanbasov commented Jul 14, 2026

Copy link
Copy Markdown
Member

Summary

Combined sync of recent development updates:

  • SafeTensors converter: accepts model B checkpoints and now fails fast when the checkpoint's code family does not match the requested config (code ≠ color), instead of producing a silently wrong export.
  • Legacy triangular color-code reference circuit restored (triangular_color_code_circuit in code/qec/surface_code/memory_circuit.py); the legacy-generator test suite passes again (45 tests, no skips).
  • README fixes: stale documentation on frame precomputation and training receptive fields; documents a known NCCL 2.28.9 cu128 SIGSEGV issue and its workaround.
  • Config fix for the d9 memory surface model 1 pre-decoder config.
  • CI stabilization: GPU smoke-training jobs now train 4 epochs instead of 2, so the LER gate (check_ler_from_log --max-ler 0.35) no longer flakes on short runs.

Aggregated from internal QA follow-ups.

Notes for reviewers

  • Deselect/ignore cleanups for the re-enabled legacy color-code tests were part of the source delta but do not apply here: this repo's workflows carry no --deselect/--ignore-glob entries for those tests, and test_generator_jax* files do not exist in this repo.
  • The multi-GPU smoke LER gate may flake once on the first CI run if it executes with the pre-fix 2-epoch config; a single rerun of that job is expected to pass.

Test plan

  • tests.test_legacy_color_code_no_overlap, tests.test_color_code_data_partition, tests.test_color_code_legacy_cnot_mapping, tests.test_color_code_homological_equivalence — 45 tests, 0 skips, all pass
  • tests.test_safetensors_export — 9 tests, all pass
  • GPU CI (copy-pr-bot gated)

🤖 Generated with Claude Code

…uit, docs and CI fixes

- SafeTensors converter: accept model B checkpoints and fail fast when the
  checkpoint's code family does not match the requested config (code != color).
- Restore the legacy triangular color-code reference circuit
  (triangular_color_code_circuit) used by the legacy-generator test suite;
  its tests pass again (45 tests, no skips).
- README: fix stale documentation (frame precomputation, training receptive
  fields) and document a known NCCL 2.28.9 cu128 SIGSEGV issue with the
  recommended workaround.
- Fix d9 memory surface model 1 pre-decoder config.
- CI: train smoke models for 4 epochs instead of 2 to stabilize the LER gate
  in GPU smoke jobs.

Aggregated from internal QA follow-ups.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@ivanbasov

Copy link
Copy Markdown
Member Author

/ok to test b32452b

@ivanbasov
ivanbasov requested review from bmhowe23 and huaweil-nv July 14, 2026 23:16
@ivanbasov
ivanbasov merged commit 79d2d1e into main Jul 15, 2026
17 checks passed
@ivanbasov
ivanbasov deleted the sync/dev-updates-2026-07-14 branch July 15, 2026 04:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants