MAINT: more unfinished work on tmjp sampler