Engineer/Senior Engineer- ML Libraries/ARM

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.)
• Memory bandwidth and cache behavior
• Parallelism and dataflow patterns
• Simulation & Trace Generation

What you need

• 2–5 yrs of experience
• Machine learning
• Python
• C++
• C
• Java
• LLMs
• B.Tech or B.E. or B.Sc or BCA or M.Tech or PhD

Good to have

• PyTorch