U.S. federal energy regulators are set to rewrite grid connection rules for data centers before June ends, potentially shifting billions in costs for AI infrastructure.