Accelerating AI Chip Development with 3D Multi-Die Designs