AdamX: Cosine similarity meets gradient descent — ThinkLLM