Skip to content

Repair parametrization notebook examples#50

Merged
lezcano merged 1 commit into
masterfrom
review/parametrization-notebook
Jun 7, 2026
Merged

Repair parametrization notebook examples#50
lezcano merged 1 commit into
masterfrom
review/parametrization-notebook

Conversation

@lezcano

@lezcano lezcano commented Jun 7, 2026

Copy link
Copy Markdown
Owner

Summary

  • fix the missing self argument in the symmetric parametrization
  • initialize nn.Module and register the pruning mask as a buffer
  • create identity matrices on the input device and dtype
  • initialize examples so native PyTorch parametrization registration succeeds

Why

Several tutorial cells were not executable with the current native PyTorch parametrization framework or after moving modules to another device or dtype.

Validation

  • compiled every notebook code cell
  • executed the repaired examples on CUDA with float64
  • validated notebook JSON

@lezcano lezcano marked this pull request as ready for review June 7, 2026 16:49
@lezcano lezcano force-pushed the review/parametrization-notebook branch from 72e11e5 to 64aba41 Compare June 7, 2026 17:00
@lezcano lezcano merged commit 1843568 into master Jun 7, 2026
9 checks passed
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.

1 participant