streamlined the collapsed network sampling. didn't really speed up inference though.