Going back to old debug mode behavior, but with fixed handling of non-broadcast "scalar" params passed to staged functions. Also putting disable_jit around method calls to prevent compilation in JAX control flow constructs.

Going back to old debug mode behavior, but with fixed handling of non-broadcast "scalar" params passed to staged functions. Also putting disable_jit around method calls to prevent compilation in JAX control flow constructs. #1513