google-research / discs

DISCS: The code base for the Benchmark for Discrete Sampling
Apache License 2.0
43 stars 7 forks source link

Dmala #48

Closed KatayoonGoshvadi closed 1 year ago

KatayoonGoshvadi commented 1 year ago

Hi Hanjun, Please find below the code for DMALA. I tested the performance for bernoulli and categorical models and they seem to match with haoran results. Note that the below PR also contains the last PR that you have approved (there were some merge issues so I re-included the changes here.)