QSGD implementation question

[tvogels]

Hi! Thanks a lot for the ATOMO paper and sharing your code over here. I just had a look at your implementation of 1-bit QSGD compression, and was wondering where the Elias coding is implemented. I would be curious if you have any thoughts on doing this efficiently. I found utils.compress, which I thought could be related, but I cannot find a use of this function in this repo.