Mnemosyne: Learning to Train Transformers with Transformers - Jain, Deepali; Choromanski, Krzysztof M; Dubey, Kumar Avinava; Singh, Sumeet; Sindhwani, Vikas; Zhang, Tingnan; Tan, Jie (Advances in Neural Information Processing Systems, 2023)