AI Frameworks Engineer: PyTorch & HPC Optimizations

Intel; Santa Clara (CA); Hybrid; USD 150,000 - 276,000; Full time

Benefits offered by this job

Hybrid work model

Stock bonuses

Health benefits

Job summary

Intel's AI Frameworks PyTorch team is seeking an AI Frameworks Engineer to design, build, and optimize AI software frameworks for scalable AI workloads. You will identify bottlenecks and implement optimizations across CPUs, GPUs, and accelerators to boost throughput and reduce latency.

The role emphasizes collaboration with Argonne National Labs, advanced model enablement, and cross-team coordination to drive robust, high-performance AI infrastructure on Intel architectures.

Qualifications

• Bachelor's degree with 3+ years of software development experience; OR Master's with 1+ year; OR Ph.D. with 3+ months of experience in software development.
• 3+ years of experience in at least one: AI framework development or optimization; PyTorch, vLLM, TensorFlow, Hugging Face, DeepSpeed; AI model development or performance optimization; GPU/accelerator software development; runtime, compiler, kernel, or backend optimization for AI workloads.

Responsibilities

• Identify performance bottlenecks and features required for Argonne AI CoE workloads.
• Optimize PyTorch, vLLM, and related AI framework software for Intel CPUs, GPUs, and accelerators.
• Enable and optimize AI models including large language models and scientific AI workloads.
• Profile AI training/inference workloads to detect issues across framework, runtime, kernel, and hardware layers.
• Collaborate with cross-functional teams to define technical specs and software requirements.
• Troubleshoot and resolve complex issues across hardware and software stacks.
• Contribute to software innovations that improve AI framework performance, scalability, and usability.
• Partner with engineering and architecture teams to maximize AI model performance on Intel architectures.

Skills

AI framework optimization

PyTorch/vLLM

Python

C++ programming

Linux environments

Performance profiling

Cross-functional collaboration

Education

Bachelor's degree in Computer Science/Engineering/STEM

Master's degree in Computer Science/Engineering/STEM

Ph.D. in Computer Science/Engineering/STEM

Tools

SYCL/CUDA

Git

Job description

Intel's AI Frameworks PyTorch team is seeking an AI Frameworks Engineer to design, build, and optimize AI software frameworks for scalable AI workloads. You will identify bottlenecks and implement optimizations across CPUs, GPUs, and accelerators to boost throughput and reduce latency.

The role emphasizes collaboration with Argonne National Labs, advanced model enablement, and cross-team coordination to drive robust, high-performance AI infrastructure on Intel architectures.