with_data shape-change fix

[elc45]

Check thoroughly if the dims of shared variables change when with_data method is called on a CompiledModel, and if so, throw an informative error. Should fix #267.

[ricardoV94]

We should support this, so if anything raise a NotImplementedError

[aseyboldt]

If the model wasn't frozen, we should be able to get away with some shape changes though? I guess we could check if the tensor type shapes match?

[ricardoV94]

If the model wasn't frozen

Doesn't nutpie freeze the model by default now?

[aseyboldt]

The default is to freeze jax models, but not numba models.
If we want to change the dataset of a model or some other shared variables it is nice if we can avoid having to recompile, and often that will involve changing the shape (of a dataset for instance) as well.

[ricardoV94]

If the model isn't frozen, changing any shared variable that doesn't have static shape (all by default), should work fine. Something in nutpie is likely hardcoding the shape after freezing though, because in my MWE I'm not using jax backend