Caleb Weinreb

Results 4 issues of Caleb Weinreb

I obtain the following error when running `train.py`. Some googling suggests this [results from a corrupt file download](https://stackoverflow.com/questions/56391392/how-to-fix-pickle-unpicklingerror-invalid-load-key-error-in-pytorch). When I run `md5sum` on `duc_se.pth`, I get `3764b858a2f612f536c841c971ea8e63` (and the file...

Currently there are associative scan implementations for hmm smoothing and filtering, but not for sampling.

This PR implements a parallel version of HMM posterior sampling using associative scan (see https://github.com/probml/dynamax/issues/341). The scan elements $E_{ij}$ are vectors specifying a sample z_j ~ p(z_j \mid z_i) for...

Thanks for making this awesome library! We are interested in optimization over the Stiefel manifold (simple gradient descent). Is this supported? If not, what is the minimal set of methods...