Standard AI coding and system design rounds with emphasis on back of the napkin calculations to understand bottlenecks and propose solutions.
Suggest going through some papers in your field and understanding challenges and sota solutions to address them.