Retired Army Gen. David Petraeus will take a new job with investment firm Kohlberg Kravis Roberts & Co. L.P. as he attempts to rebuild his reputation after an extramarital affair with a biographer triggered his resignation as CIA director last fall.