pyro-ppl / numpyro / 13773802920

92%

push

github

Add `CirculantNormal` distribution. (#1988)

* Add `CirculantNormal` distribution.

* Fix Jacobian of `RealFastFourierTransform`.

* Add transform to pack and unpack Fourier coefficients.

* Implement `CirculantNormal` using transformations.

* Fix typo in skip reason for `test_generated_sample_distribution`.

* Add `CirculantNormal` example notebook.

* Reformat `circulant_gp` example notebook.

* Fix keyword argument for `PackRealFastFourierCoefficientsTransform` test.

* Fix batch shape for mean, covariance_row, covariance_matrix.

* Add analytic KL divergence of diagonal `Normal` from `CirculantNormal`.

* Fix sign bug in `_PositiveDefiniteCirculantVector` validity check.

* Expand docstring and add references for `Circulant` distribution.

* Fix shape promotion for `Circulant` distribution args.

* Skip goodnes of fit test for `CirculantNormal`.

* Expand docstring of `CirculantNormal`.

* Update example and move legend.

---------

Co-authored-by: Du Phan <fehiepsi@gmail.com>

118 of 120 new or added lines in 5 files covered. (98.33%)

13590 of 14627 relevant lines covered (92.91%)

1.98 hits per line