xarray-einstats
xarray-einstats copied to clipboard
arviz-devs
Stats, linear algebra and einops for xarray
Would it be possible to add numpy.linalg.pinv functionality please? Looks like it should be very similar to inv.
The type info currently there doesn't make much sense: https://github.com/arviz-devs/xarray-einstats/blob/main/src/xarray_einstats/numba.py#L13-L22
It would be nice to be able to do things like ``` with matrix_dims(["dim1", "dim3"]): chol = xe.linalg.cholesky(da) eig = xe.linalg.eig(da) ``` instead of having to use the monkeypatch trick...
PreliZ distributions are similar to scipy ones in many aspects, but sometimes it is more convenient to work with them. Some differences are that both continues and discrete PreliZ's distributions...
``` ========================================================================================= FAILURES ========================================================================================== __________________________________________________________________________________ TestWrappers.test_pinv ___________________________________________________________________________________ self = matrices = Size: 4kB 0.11 0.3897 1.4 2.2 0.3435 0.2578 0.4345 ... 0.3557 1.105 2.256 2.58 1.683 1.311 Dimensions without coordinates:...
Would distance metrics like wasserstein and kl-divergence (ok ok it's not a proper distance) be within the scope of this project? It would be nice to have an out-of-the-box way...
