Update missing modalities imputation. - [ ] Update code to work with the latest changes in decoders - [ ] Add uncertainty estimation - [ ] Add tutorial
