Yes, Martin Amis did write "Einstein's Monsters." The short story collection deals with the threat of nuclear war.