No he didn't. He understood the science - and wrote a highly-public letter to the president arguing it would be better for the US to build one than to have it dropped on them by the Germans - already at work. But he had no hand in building it. Keep in mind, Einstein would have been in his mid-60s during the war, and a German at that.