[dev] Work on nuclei_densmat.