-
-
Notifications
You must be signed in to change notification settings - Fork 66
Description
Is your feature request related to a problem? Please describe.
This isn't a big deal, but now that we have multiple sampling algorithms, it would be nice if we could condense the syntax for specifying multi-threading. Currently, the sample method has threads_per_chain, pathfinder has num_threads, and laplace has threads. Especially as some of the utility of the variational methods is in using them as testing or initialization for MCMC, it would be nice if we could unify this syntax.
Describe the solution you'd like
Perhaps just a single argument like num_threads or perhaps for compatibility with sample the others could all use threads_per_chain.
Describe alternatives you've considered
Perhaps the other variational methods could also take threads_per_chain as an additional argument if threads or num_threads isn't specified.
Additional context
This is mainly for using cmdstanr in a pre-compiled model context in which switching from one sampling method to another may be done for the same data/model.