We have a lot of "check if scalar" code in filters/simulators, and the jaxtyping should inspire us to handle this at the highest level (at DynMod init and sample init) so that everything downstream can rely on single shape conventions. Can do this in tandem with a full jaxtype overhaul of Simulators.py
We have a lot of "check if scalar" code in filters/simulators, and the jaxtyping should inspire us to handle this at the highest level (at DynMod init and sample init) so that everything downstream can rely on single shape conventions. Can do this in tandem with a full jaxtype overhaul of Simulators.py