> anything else being wiped on reboot

Have you tried Impermanence?

https://nixos.wiki/wiki/Impermanence