chore: split kpms into separate optional dependencies for CPU and GPU, update README

chore: split kpms into separate optional dependencies for CPU and GPU, update README #185