2017 Freightliner cascadia evo automatic transmission stalling.

My 2017 Freightliner Cascadia evo has 600k on the dash recently it has been occasionally stalling in neutral not going down the highway but in city travel or coming off the highway after a complete stop or when I try to automatically shift it into like 2nd or 3rd not sure what exactly it is the transmission which is still under warranty till 750k or the clutch or something else