Oppenheimer, Einstein y la bomba atómica