While that’s true, I would suggest using CustomDist as it will play nicer with dims/shapes (it will feel like using a vanilla distribution). model.register_rv is more internal and was used mostly as a work-around while CustomDist was being revamped.
We don’t promise to maintain the register_rv API stable, while we will try to do that for CustomDist