STEM OPT AI Research Engineer Jobs
AI Research Engineer roles sit squarely within STEM OPT eligibility, letting you work for up to 36 months post-graduation without H-1B visa sponsorship. Your employer must be enrolled in E-Verify, and your degree in computer science, electrical engineering, or a related STEM field must align with the role's CIP code to qualify for the 24-month extension.
Find STEM OPT AI Research Engineer JobsOverview
Showing 5 of 426+ AI Research Engineer jobs










See all 426+ AI Research Engineer Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new AI Research Engineer roles.
Get Access To All Jobs
SUMMARY
The Multimodal Intelligence Team is building the next generation of foundation models for Apple experiences. We are looking for a research scientist to advance the architectures, pre-training methods, and distillation techniques that make highly capable multimodal models practical across the Apple ecosystem. Our research spans the full foundation-model lifecycle: model architecture, pre-training objectives, data mixtures, optimization, scaling, distillation, and evaluation. A defining challenge of our work is to develop models that combine broad intelligence with the memory, latency, energy, and privacy requirements of on-device deployment. You will have the opportunity to shape new research directions, conduct ambitious experiments at scale, and translate successful ideas into foundation-model technologies that can reach Apple products. Where appropriate, this work may also lead to publications and the open sourcing of selected models, research artifacts, evaluations, or tools.
DESCRIPTION
In this role, you will investigate fundamental questions about how multimodal foundation models should be designed, trained, and distilled. You will develop and evaluate new model architectures, pre-training objectives, data strategies, optimization methods, and teacher–student learning techniques. Your work will explore how capabilities developed in large foundation models can be effectively transferred to smaller, more efficient models without treating distillation as an isolated downstream step. A major focus of the role will be the co-development of frontier models and efficient models for Apple silicon and on-device intelligence. This includes designing architectures that distill effectively, studying how teacher and student models should be trained together, and developing distillation methods that preserve reasoning, multimodal understanding, instruction following, and other important capabilities under constrained model capacity. Rather than treating deployment constraints as an afterthought, you will incorporate them into the research process—from early architecture experiments and pre-training through distillation and final model evaluation. You may thrive in this role if you:
- Want to invent new foundation-model architectures rather than only adapt existing models.
- Enjoy combining scientific ambition with real compute, memory, latency, and energy constraints.
- Believe that small and efficient models can be a frontier research problem, not merely a compression exercise.
- Are comfortable working across model research, data, systems, and hardware boundaries.
- Care about translating research into private, useful, and deeply integrated intelligent experiences.
- Want your work to have both product impact and a presence in the broader research community.
Potential research directions include:
- Novel dense, recurrent, state-space, mixture-of-experts, and hybrid foundation-model architectures.
- Multimodal pre-training across language, images, video, audio, and sensor-derived representations.
- Compute-optimal model and data scaling, including data mixtures, curricula, tokenization, and training objectives.
- Architecture and algorithm co-design for memory-efficient and energy-efficient inference on Apple silicon.
- Offline and on-policy distillation using teacher-generated data, logits, representations, rationales, and other supervision signals.
MINIMUM QUALIFICATIONS
- Hands-on experience designing, implementing, and running large-scale pre-training experiments for large language models.
- Experience with LLM pre-training topics such as model architecture, training objectives, data mixtures, tokenization, curricula, scaling, and optimization.
- Strong proficiency with modern deep learning frameworks such as PyTorch or JAX and distributed training systems.
- Experience evaluating pre-trained models across language understanding, reasoning, instruction following, or multimodal capabilities.
- Strong understanding of transformer-based architectures and current approaches to efficient or scalable foundation-model training.
- Master’s degree, or equivalent practical experience in machine learning, computer science, or a related technical field.
PREFERRED QUALIFICATIONS
- Experience contributing to major foundation-model pre-training efforts or leading architecture experiments that influenced a large training run.
- Research contributions in model architecture, scaling laws, multimodal pre-training, optimization, efficient attention, mixture-of-experts, state-space models, or related areas.
- Experience with knowledge distillation, including offline or off-policy distillation, on-policy distillation, self-distillation, sequence-level distillation, logic matching, or representation transfer.
- Experience designing teacher–student training pipelines or transferring capabilities from large foundation models to smaller models.
- Experience with multimodal models spanning language, vision, video, audio, or other sensor modalities.
- Understanding of inference efficiency, memory hierarchy, hardware accelerators, or hardware–software co-design.
- Strong publication record, influential open-source contributions, or an equivalent record of applied research impact.
COMPENSATION
- Base pay range for this role is between $150,400 and $277,600, and your base pay will depend on your skills, qualifications, experience, and location.
Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation.
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.
At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.
Apple accepts applications to this posting on an ongoing basis.
See all 426+ STEM OPT AI Research Engineer Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new STEM OPT AI Research Engineer Jobs.
Get Access To All JobsTips for Finding STEM OPT Authorization as an AI Research Engineer
Verify your CIP code before applying
Your STEM OPT extension depends on your degree's CIP code matching a STEM-designated field. Check your I-20 with your DSO before you start applying to confirm your program qualifies, not after you've accepted an offer.
Screen employers for E-Verify enrollment early
STEM OPT employers must be enrolled in E-Verify before your extension starts. Ask your recruiter during the first interview, not at the offer stage. A company without E-Verify enrollment cannot legally employ you on STEM OPT.
Target labs with established research training structures
The I-983 training plan requires a specific learning objective tied to your STEM degree. AI research roles at companies or universities with existing research divisions are easier to document because training plans there map directly to defined project scopes.
Use OFLC Wage Search to benchmark your offer
AI Research Engineer wages vary significantly by employer type and geography. Pull the prevailing wage for SOC code 15-2051 or 15-1299 from OFLC Wage Search before negotiating so you know whether an offer meets DOL wage standards for your location.
Find STEM OPT-ready employers on Migrate Mate
Search AI Research Engineer roles on Migrate Mate, which filters employers by E-Verify enrollment and STEM OPT compatibility. This saves you from applying to companies that can't legally support your work authorization before the conversation even starts.
File your STEM OPT extension 90 days before OPT expires
USCIS requires your extension application to be received before your initial OPT EAD expires. File at least 90 days out to stay within the 180-day filing window and avoid a gap in work authorization between initial OPT and your extension approval.
Frequently Asked Questions
Does my STEM degree qualify me for the STEM OPT extension as an AI Research Engineer?
Your degree qualifies if it's in a STEM-designated field and your program's CIP code appears on the DHS STEM Designated Degree Program List. Computer science, electrical engineering, mathematics, and statistics are common qualifying fields for AI Research Engineer roles. Your DSO can confirm your specific CIP code matches before you apply for the 24-month extension.
What does the I-983 training plan require for an AI Research Engineer role?
The I-983 requires your employer to document how the AI Research Engineer role provides practical training directly related to your STEM degree. This means listing specific learning objectives, the skills you'll develop, and how they connect to your field of study. Your employer and DSO must both sign the form, and it must be on file before your extension starts.
How do I confirm my employer is enrolled in E-Verify before accepting an offer?
Ask your recruiter or HR contact directly whether the company is currently enrolled in E-Verify as an employer, not just a participant. You can also search the E-Verify employer search tool by company name. E-Verify enrollment is a hard legal requirement for STEM OPT, and enrollment must be active at the time your extension begins, not just when you were hired.
How do I find AI Research Engineer employers that support STEM OPT?
Migrate Mate filters AI Research Engineer listings by E-Verify enrollment and STEM OPT eligibility, so you're not manually vetting each company. This matters because many job postings don't disclose E-Verify status upfront, and finding out after an offer wastes time for both you and the employer.
What happens to my STEM OPT authorization if my employer loses E-Verify enrollment?
If your employer's E-Verify enrollment terminates while you're on STEM OPT, your work authorization is no longer valid at that employer. USCIS requires you to notify your DSO within five days of losing employment. You then have a 60-day grace period to find a new qualifying employer, transfer your STEM OPT, or change status. Staying employed beyond that point without a valid authorization is a status violation.