Skip to content

Commit 5222e82

Browse files
committed
Vectorize parameter-shift gradient hot path and cache global-Z eigenvalues
Replace Python loops in the trainability inner loop with cached and fully vectorised helpers in encoding_atlas.analysis._utils, and batch compute_all_parameter_gradients into one simulator dispatch instead of 2 * n_features independent calls. Gradient values are numerically identical to the previous per-call implementation.
1 parent 27c14fd commit 5222e82

4 files changed

Lines changed: 942 additions & 209 deletions

File tree

0 commit comments

Comments
 (0)