Why the route returns through Chiang Mai
The clearest public-transport sequence is Chiang Mai–Pai–Chiang Mai–Chiang Rai. Trying to cut directly from Pai to Chiang Rai creates a long cross-region road day and fewer recovery options if a vehicle or weather problem intervenes.
TAT identifies Highway 1095 as the mountain route to Pai. Treat each Pai leg as real travel time and keep the day after arrival light if winding roads affect you.