is the new unimax sampling based mixture implementation available ? from: https://openreview.net/pdf?id=kXwdL1cWOAi