AdamX: Cosine similarity meets gradient descent · Bharat Hunt