Adaptive Sampling for Efficient Softmax Approximation - Baharav, Tavor Z.; Kang, Ryan; Sullivan, Colin; Tiwari, Mo; Luxenberg, Eric; Tse, David; Pilanci, Mert (Advances in Neural Information Processing Systems, 2024)