arxiv Learning to Retrieve In-Context Examples for Large Language Models