PE- CPU Software & Hardware Co-Design Engineer (ML Systems)

About the role

CPU Software & Hardware Co-Design Engineer (ML Systems) This role focuses on CPU software–hardware co-design for next-generation QMX architectures , including workload characterization, simulation, kernel optimization, and driving architectural insights for future CPU designs. The ideal candidate will work across the full stack—from ML models to low-level kernels to architectural feedback—enabling efficient execution of ML workloads on CPU platforms .

What you'll do

• ML Workload Identification & Characterization
• Identify and prioritize critical ML use cases and models for CPU-centric execution (LLMs, vision, speech, recommender systems, etc.)
• Analyze workload characteristics including: Compute intensity
• Memory bandwidth and cache behavior
• Parallelism and dataflow patterns

What you need

• 8–14 yrs of experience
• Machine learning
• Hardware
• Java
• Python
• C++
• C
• B.Tech or B.E. or B.Sc or BCA or M.Tech or PhD

Good to have

• LLMs
• PyTorch