Fix Blackjax SMC integration

[ciguaran]

Fixes #354

[ciguaran]

@ricardoV94 do you have any insight on pytensor's E AttributeError: A config parameter with the name 'floatX' was already registered on another config instance. ? I can't seem to reproduce it locally with python 3.10. Locally all tests pass.

[ricardoV94]

@ciguaran I don't think it's related to your changes. Perhaps something @Armavica changed in PyTensor recently?

[Armavica]

We talked about it with @ferrine some time ago. It started happening around the time of #361 so it could be related to that too.
The error happens on my computer too. I remember I tried bisecting pymc-experimental, including downgrading pytensor, but didn't manage to understand what caused it.

[ricardoV94]

Rebasing may magically solve the failures now

[ciguaran]

@ricardoV94 it seems so!

[ricardoV94]

Thanks @ciguaran !