MAINT: move chain-specific sampling functions into their own module to distinguish them from sampling functions that require edge weights