Make contracts explicit for frequently used Callables

[ciguaran]

Although some interfaces depend on frequently used concepts, such as logprob_fn or root solver, these parameters are all typed just as Callables. This makes it difficult for users to really understand what type of function they should use. We need to make expectations on these Callables explicit, possibly by refining the type annotations.