Commit 5222e82
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
- src/encoding_atlas/analysis
- tests/unit/analysis
0 commit comments