Gene co-expression in single-cell RNA-seq

I am using single-cell RNA-seq data from the Allen Institute, and I want to look at gene co-expression in different cell populations. They provide raw UMI counts, so I’m wondering what normalization method to use (e.g., CPM, TPM, VST) to look at these correlations. Any rationale/justification is appreciated.