AI Frameworks Engineer

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

• AI model development, enablement, profiling, or performance optimization.

Job Details

Job Description:

Join Intel's AI Frameworks PyTorch team to shape the future of Artificial Intelligence (AI) and High-Performance Computing (HPC). As an AI Frameworks Engineer, you will design, build, and optimize AI software frameworks that enable breakthrough scientific discoveries and machine learning innovations at scale.

You will contribute to the success of the new Argonne AI Center of Excellence, a partnership between Intel and Argonne National Labs. This role shall focus on identifying performance bottlenecks and functional limitations in PyTorch, vLLM, and related AI framework software when running key AI workloads and models at large scale. After identifying issues that impact functionality, scalability, and performance, you will help design and implement optimizations that maximize the utility of Intel CPUs, GPUs, and AI accelerators.

You will work as part of a larger team developing and optimizing next-generation AI framework software, enabling efficient training, inference, and deployment of modern AI models. Your innovative work will improve performance across Intel architectures and empower HPC and AI applications to achieve higher throughput, lower latency, and stronger developer productivity.

Key Responsibilities

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

Qualifications

You must possess the below minimum qualifications to be initially considered for this position. Preferred qualifications are in addition to the requirements and are considered a plus factor in identifying top candidates.

Minimum Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, Mathematics, or STEM-related field with 3+ yrs. of experience in software development. OR
• Master's degree in Computer Science, Computer Engineering, Electrical Engineering, Mathematics, or STEM-related field with 1+ yrs. of experience in software development. OR
• Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, Mathematics, or STEM-related field with 3+ months of experience in software development.
• 3+ years of experience in at least one of the following:
• AI framework development or optimization.
• PyTorch, vLLM, TensorFlow, Hugging Face, DeepSpeed, or related AI software frameworks.
• AI model development, enablement, profiling, or performance optimization.
• GPU or accelerator software development.
• Runtime, compiler, kernel, or backend optimization for AI workloads.

Preferred Qualifications

• Advanced degree, Master's or PhD, in Computer Science, Computer Engineering, Electrical Engineering, Mathematics, or STEM-related field.
• Proficiency in Python, SYCL/CUDA, and C++ programming.
• Experience developing in Linux environments.
• Background in AI framework internals, model execution, or backend integration.
• Experience with PyTorch, vLLM, Hugging Face Transformers, DeepSpeed, or similar AI frameworks.
• Experience optimizing AI training or inference workloads for CPUs, GPUs, or accelerators.
• Background in performance profiling, memory optimization, throughput improvement, or latency reduction.
• Experience enabling or optimizing large language models, generative AI models, or scientific AI models.
• Understanding of deep learning algorithms, model architectures, and AI workload patterns.
• Strong analytical skills and ability to solve complex software challenges.
• Passion for driving meaningful advancements in AI software and scientific computing.

Job Type

Experienced Hire

Shift

Shift 1 (United States of America)

Primary Location

US, California, Santa Clara

Additional Locations

US, Oregon, Hillsboro, US, Texas, Austin

Posting Statement

All qualified applicants will receive consideration for employment without regard to race, color, religion, religious creed, sex, national origin, ancestry, age, physical or mental disability, medical condition, genetic information, military and veteran status, marital status, pregnancy, gender, gender expression, gender identity, sexual orientation or any other characteristic protected by local law, regulation, or ordinance.

Position of Trust

N/A

Benefits

We offer a total compensation package that ranks among the best in the industry. It consists of competitive pay, stock bonuses, and benefit programs which include health, retirement, and vacation. Find out more about the benefits of working at Intel.

Annual Salary Range for jobs which could be performed in the US: $149,750.00 - 275,580.00 USD

The range displayed on this job posting reflects the minimum and maximum target compensation for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific compensation range for your preferred location during the hiring process.

Work Model for this Role

This role will be eligible for our hybrid work model which allows employees to split their time between working on-site at their assigned Intel site and off-site. * Job posting details (such as work model, location or time type) are subject to change.

ADDITIONAL INFORMATION

Intel is committed to Responsible Business Alliance (RBA) compliance and ethical hiring practices. We do not charge any fees during our hiring process. Candidates should never be required to pay recruitment fees, medical examination fees, or any other charges as a condition of employment. If you are asked to pay any fees during our hiring process, please report this immediately to your recruiter.