Imported from https://github.com/umutsimsekli/sgd_tail_index