Initial Release - AI-based pre-decoders for surface codes
This is an open-source release of the training framework and inference framework used for [1] and [2]. See the README.md and cookbook/predecoder.ipynb for full details on how to explore the current implementation and begin extending with your own noise models. This release can be paired with the fast and accurate models on Hugging Face!
Additionally, this can be paired with CUDA-Q QEC 0.6.0 to run a trained decoder on NVQLink!
[1] https://research.nvidia.com/publication/2026-04_fast-ai-based-pre-decoders-surface-codes
[2] https://arxiv.org/abs/2604.12841