refactor: Improve PSD typing#522

Merged

ValerianRey merged 37 commits intomainfrom

add-generalized-matrix-psd-matrix

Jan 23, 2026

Commits on Jan 20, 2026

fix is_psd_quadratic_form
PierreQuinton
committed
Rename `PSDQuadraticForm` to `PSDGeneralizedMatrix`
PierreQuinton
committed
fix type of weighting in Flattening
PierreQuinton
committed
Add parametrization of zero matrix for test_gramian_is_psd
PierreQuinton
committed
Add test of the PSD property for functions in aggregation/_utils/gramian
PierreQuinton
committed
rename test of equivariance accordingly
PierreQuinton
committed
Rename functions in `autogram/_gramian_utils` so that they don't include gramian, they are not about Gramians but about PSD(Generalized)Matrices
PierreQuinton
committed
Test the PSD property on outputs of functions in `autogram/_gramian_utils`
PierreQuinton
committed
Remove internal checks of shapes of matrices
PierreQuinton
committed
Remove uninformative shadowing of assertion error in assert_psd_*
PierreQuinton
committed
Factorize `compute_gramian` from `forward_backward` by making the one in _linalg take a GeneralizedMatrix. Update test accordingly.
PierreQuinton
committed
Revert "Factorize `compute_gramian` from `forward_backward` by making the one in _linalg take a GeneralizedMatrix. Update test accordingly."
PierreQuinton
committed
Generalizes `compute_gramian` to take a `GeneralizedMatrix` instead.
PierreQuinton
committed
Move `aggregation/_utils/gramian.py` to `_linalg/gramian.py`
PierreQuinton
committed
Merge branch 'main' into add-generalized-matrix-psd-matrix
ValerianRey
authored

Commits on Jan 22, 2026

Can parametrize number of dimensions to contract in `compute_gramian`
PierreQuinton
committed