Thanks! I linked the paper. They just defined the loss on the sampling, but did not provide code, so I am not sure how they did it.
I do not know what is ‘a differentiable model to do the sampling’. Could you give me more details on this?