Balancing hardware capabilities between edge and cloud for AI Compute