Tip: install Pop!_OS - Nvidia edition. I also have a (1st gen) Thinkpad X1 Extreme with a 4K screen. In addition to that, I have 3 external 4K screens connected to it (so 4x 4K screens in total), and it runs smoothly without having needed any additonal config/customization.