Liquid cooling for AI workloads