__init__
scalable_softmax
