AI Software Solutions Engineer

Overview

Optimize AI solutions on Intel products by working with open-source frameworks such as PyTorch, TensorFlow and JAX, within Intel's Data Center Group.

What you'll do

• Propose optimizations in software and kernel implementation to improve workload performance.
• Model AI hardware and software (GPU/accelerator) for performance analysis.
• Analyze AI workloads to identify critical kernels and patterns.
• Estimate ideal performance of kernels and workloads for a given hardware spec.
• Enhance and maintain the performance modeling toolkit for projections.
• Correlate projected performance with device/simulator and improve models.
• Study new and different deep learning networks for future optimization requirements.
• Design and develop hardware-agnostic and hardware-aware software features for AI frameworks, focusing on ML kernel development.
• Enhance and extend deep learning training and inference capabilities within Intel's AI software stack.
• Debug and resolve functional and convergence issues in multi-layered software systems.

What you'll need

• Bachelor's degree with 8+ years, Master's degree with 6+ years, or PhD with 3+ years of relevant industry experience in Computer Science, Computer Engineering or Electrical Engineering or a related technical discipline.
• Programming in Python/C++.
• Strong in data structures and algorithms.
• Experience working on large software projects.
• Knowledge of computer architecture and compiler fundamentals.
• Strong written and verbal communication skills.
• Expertise in performance optimization techniques for AI workloads on modern hardware architectures.

Nice to have

• Familiarity with AI algorithms (e.g. LLMs) and deep learning and serving frameworks (e.g. PyTorch, JAX, vLLM, SGLang, HF).
• Advanced understanding of machine learning concepts and deep learning practices.
• Knowledge of Intel hardware architecture and features, including CPUs, GPUs, or AI accelerators.
• Demonstrated ability to analyze, prototype, and solve complex technical challenges.
• Strong communication skills to convey technical concepts to diverse audiences.
• Experience developing and integrating CUTLASS or Triton-based kernels in large language models (LLMs).
• Knowledge of compiler algorithms for heterogeneous systems and fuser optimizations.
• Effective communication skills and experience working in cross-geo teams.
• Familiarity with generative AI models and LLMs.

Details

• Location: India, Bangalore.
• Work model: hybrid, splitting time between on-site at the assigned Intel site and off-site.
• Shift: Shift 1 (India).
• Job type: Experienced Hire.