Unimax sampler implementation?

I tried searching the code for seqio mixtures generated using the newly released unimax sampler. 

I am trying to pretrain a custom umUL2 model. if i could perhaps know how to implement unimax it would be of great help. Thanks