LLM memory distillation reinforcement learning