New Compute Bench

[Aaron-Zhao123]

NewComputeBench Integration - Task List

Goal

We would like to add all NewComputeBench related code change in a single PR.

---

Core Module Transformation Passes

For each module-level transformation pass on New Compute Paradigm, complete the following:

SNN pass (SpikeZIP-TF @JeffreyWong20 ):

• Rename model name to snn rather than ann2snn
• Implement new module transform pass
  • Integrate into existing pipeline
• Documentation
  • Write tutorial (under docs/tutorials/newcompute)
  • Add to main documentation (docs/source/modules/documentation)
  • Add module-level documentation (docs/source/modules/chop/module_transform)
• Tests
  • Unit tests
  • Integrate a minimal test in current CI
• Result reporting on project website

SNN pass (STA @GeorgeWu1204 ):

• Implement new module transform pass
  • Integrate into existing pipeline
• Documentation
  • Write tutorial (under docs/tutorials/newcompute)
  • Add to main documentation (docs/source/modules/documentation)
  • Add module-level documentation (docs/source/modules/chop/module_transform)
• Tests
  • Unit tests
  • Integrate a minimal test in current CI
• Result reporting on project website

ONN pass (STA @ChengZhang-98):

• Implement new module transform pass
  • Integrate into existing pipeline
• Documentation
  • Write tutorial (under docs/tutorials/newcompute)
  • Add to main documentation (docs/source/modules/documentation)
  • Add module-level documentation (docs/source/modules/chop/module_transform)
• Tests
  • Unit tests
  • Integrate a minimal test in current CI
• Result reporting on project website

PIM pass (PIM @firemountain154B ):

• Implement new module transform pass
  • Integrate into existing pipeline
• Documentation
  • Write tutorial (under docs/tutorials/newcompute)
  • Add to main documentation (docs/source/modules/documentation)
  • Add module-level documentation (docs/source/modules/chop/module_transform)
• Tests
  • Unit tests
  • Integrate a minimal test in current CI
• Result reporting on project website

---

Documentation & Testing Review (@Binglei @przemekff)

• Review and update existing tests (pytest)
  • Verify all tests pass (if not skipped)
  • Fix broken tests
  • Add missing test coverage
• Review and update tutorials
  • Verify all tutorials work end-to-end
  • Update outdated content
  • Add missing examples
• Review current documentation
  • Fix errors and inconsistencies
  • Update for new features

---

Optimizations (To Consider, @przemekff)

• Package Management Migration
  • Evaluate UV-based package management
  • Create migration plan if approved
  • Implement migration
  • Update documentation
• Code Formatting
  • Migrate from Black to Ruff
  • Configure Ruff for automated reformatting
  • Apply formatting to entire codebase
  • Update CI/CD pipelines
  • Update contributor guidelines