STEM OPT Senior ML Engineer Jobs
Senior ML Engineer roles qualify for the 24-month STEM OPT extension when your degree falls under an eligible CIP code and your employer is enrolled in E-Verify. With 36 months of total OPT work authorization, you have a realistic runway to build production ML experience and secure H-1B visa sponsorship.
Find STEM OPT Senior ML Engineer JobsOverview
Showing 5 of 621+ Senior ML Engineer jobs
See all 621+ Senior ML Engineer Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new Senior ML Engineer roles.
Get Access To All JobsDESCRIPTION
Annapurna Labs was a startup company acquired by AWS in 2015, and is now fully integrated. If AWS is an infrastructure company, then think Annapurna Labs as the infrastructure provider of AWS. Our org covers multiple disciplines including silicon engineering, hardware design and verification, software, and operations. AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last few years.
AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the Trn1 and Inf1 servers that use them. This position is for a Software Engineer that will lead the development of machine learning tools to run, optimize, and analyze machine learning workloads. This candidate must have had experience leading machine learning tool projects, preferably starting from architecture through several generations of delivery to customers. Deep knowledge of profiling and optimization, resource management, scheduling, code generation are needed. The ideal candidate will have worked on new instruction set architectures, which may include CPU, NPU, GPU and other forms of compute.
Key job responsibilities
This engineer will lead the design and implementation of ML infrastructure platform, building systems for capacity management, workload scheduling, and fleet orchestration across ML accelerators. They will work with ML scientists, training infrastructure engineers, hardware teams, and internal customers to ensure the ML Infra service delivers seamless ML Accelerator access with low wait times, high utilization, and zero-config deployment from various environments.
A day in the life
As you design and code solutions to help our team drive efficiencies in software architecture, you’ll create metrics, implement automation and other improvements, and resolve the root cause of software defects. You’ll also:
- Build high-impact solutions to deliver to our large customer base.
- Participate in design discussions, code review, and communicate with internal and external stakeholders.
- Work cross-functionally to help drive business decisions with your technical input.
- Work in a startup-like development environment, where you’re always working on the most important stuff.
About the team
- High-impact, high-visibility: You'll directly accelerate every Neuron team's ability to ship — your work multiplies the output of 100+ engineers
- Greenfield opportunities: We're actively building new capabilities with significant design ownership for SDEs
- Small, senior team: where every person owns major components and drives architectural decisions
- AI infrastructure: Work at the intersection of Kubernetes, custom silicon, and large-scale ML workloads
Diverse Experiences
We value diverse experiences and non-traditional career paths. If your career is just starting or includes alternative experiences, we encourage you to apply.
Inclusive Team Culture
Our employee-led affinity groups foster inclusion. Events like CORE and AmazeCon inspire us to embrace our uniqueness.
Work/Life Balance
We strive for flexibility as part of our working culture, supporting you both at work and at home.
Mentorship & Career Growth
We offer knowledge-sharing, mentorship, and one-on-one code reviews to help you grow as a professional.
BASIC QUALIFICATIONS
- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
PREFERRED QUALIFICATIONS
- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent
- 2+ years of building large-scale machine-learning infrastructure for online recommendation, ads ranking, personalization or search experience
- Strong proficiency in Go/Java, Python and working knowledge Javascript/TypeScript
- Experience building and operating large-scale distributed systems on Kubernetes
- Experience designing, deploying, and maintaining production services at scale, including on-call ownership
- Experience with machine learning infrastructure — orchestration, scheduling, or resource management at scale
- Proficiency in application and kernel-level performance profiling and optimization
- Experience with integrated software/hardware performance analysis in heterogeneous compute environments
- Proficiency in observability and telemetry — instrumentation, metrics collection, alarming, dashboarding, and monitoring
- Experience debugging complex issues in large-scale distributed systems and driving best practices
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
LOCATION
USA, WA, Seattle - 143,700.00 - 194,400.00 USD annually
DESCRIPTION
Annapurna Labs was a startup company acquired by AWS in 2015, and is now fully integrated. If AWS is an infrastructure company, then think Annapurna Labs as the infrastructure provider of AWS. Our org covers multiple disciplines including silicon engineering, hardware design and verification, software, and operations. AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last few years.
AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators and the Trn1 and Inf1 servers that use them. This position is for a Software Engineer that will lead the development of machine learning tools to run, optimize, and analyze machine learning workloads. This candidate must have had experience leading machine learning tool projects, preferably starting from architecture through several generations of delivery to customers. Deep knowledge of profiling and optimization, resource management, scheduling, code generation are needed. The ideal candidate will have worked on new instruction set architectures, which may include CPU, NPU, GPU and other forms of compute.
Key job responsibilities
This engineer will lead the design and implementation of ML infrastructure platform, building systems for capacity management, workload scheduling, and fleet orchestration across ML accelerators. They will work with ML scientists, training infrastructure engineers, hardware teams, and internal customers to ensure the ML Infra service delivers seamless ML Accelerator access with low wait times, high utilization, and zero-config deployment from various environments.
A day in the life
As you design and code solutions to help our team drive efficiencies in software architecture, you’ll create metrics, implement automation and other improvements, and resolve the root cause of software defects. You’ll also:
- Build high-impact solutions to deliver to our large customer base.
- Participate in design discussions, code review, and communicate with internal and external stakeholders.
- Work cross-functionally to help drive business decisions with your technical input.
- Work in a startup-like development environment, where you’re always working on the most important stuff.
About the team
- High-impact, high-visibility: You'll directly accelerate every Neuron team's ability to ship — your work multiplies the output of 100+ engineers
- Greenfield opportunities: We're actively building new capabilities with significant design ownership for SDEs
- Small, senior team: where every person owns major components and drives architectural decisions
- AI infrastructure: Work at the intersection of Kubernetes, custom silicon, and large-scale ML workloads
Diverse Experiences
We value diverse experiences and non-traditional career paths. If your career is just starting or includes alternative experiences, we encourage you to apply.
Inclusive Team Culture
Our employee-led affinity groups foster inclusion. Events like CORE and AmazeCon inspire us to embrace our uniqueness.
Work/Life Balance
We strive for flexibility as part of our working culture, supporting you both at work and at home.
Mentorship & Career Growth
We offer knowledge-sharing, mentorship, and one-on-one code reviews to help you grow as a professional.
BASIC QUALIFICATIONS
- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
PREFERRED QUALIFICATIONS
- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent
- 2+ years of building large-scale machine-learning infrastructure for online recommendation, ads ranking, personalization or search experience
- Strong proficiency in Go/Java, Python and working knowledge Javascript/TypeScript
- Experience building and operating large-scale distributed systems on Kubernetes
- Experience designing, deploying, and maintaining production services at scale, including on-call ownership
- Experience with machine learning infrastructure — orchestration, scheduling, or resource management at scale
- Proficiency in application and kernel-level performance profiling and optimization
- Experience with integrated software/hardware performance analysis in heterogeneous compute environments
- Proficiency in observability and telemetry — instrumentation, metrics collection, alarming, dashboarding, and monitoring
- Experience debugging complex issues in large-scale distributed systems and driving best practices
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
LOCATION
USA, WA, Seattle - 143,700.00 - 194,400.00 USD annually
See all 621+ STEM OPT Senior ML Engineer Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new STEM OPT Senior ML Engineer Jobs.
Get Access To All JobsTips for Finding STEM OPT Authorization as a Senior ML Engineer
Verify your CIP code before applying
Check your degree's CIP code against the DHS STEM OPT designated degree list before targeting Senior ML Engineer roles. Computer science, electrical engineering, and statistics degrees typically qualify, but applied mathematics subfields vary by program classification.
Filter jobs by E-Verify enrollment status
Senior ML Engineer postings don't always disclose E-Verify enrollment. Run the employer name through the E-Verify employer search before accepting a screening call so you don't spend three rounds of interviews on an ineligible company.
Frame your ML specialization in LCA job duty language
When reviewing offer letters, confirm the job duties reference model development, training pipelines, or deployment infrastructure. DOL LCA filings for Senior ML Engineers that use vague duty language face higher OFLC scrutiny, which can delay your start date.
Draft your I-983 training plan before the offer stage
Prepare a draft I-983 that maps your STEM OPT training objectives to specific ML engineering deliverables like model optimization or MLOps workflows. Employers unfamiliar with STEM OPT move faster when you arrive with a near-complete plan rather than explaining the form from scratch.
Use Migrate Mate to target employers with STEM OPT hiring history
Search Senior ML Engineer roles on Migrate Mate, which surfaces employers filtered for E-Verify enrollment and prior STEM OPT and H-1B filing activity. This narrows your list to companies already set up for the compliance steps your authorization requires.
Time your STEM OPT application around H-1B cap registration
If your initial OPT expires before October 1, file your STEM OPT extension with your DSO at least 90 days early so cap-gap coverage keeps you authorized through the H-1B start date. USCIS requires the extension application to be timely filed for cap-gap to apply.
Frequently Asked Questions
Does a Senior ML Engineer role qualify for the STEM OPT extension?
Yes, if your degree is in a DHS-designated STEM field such as computer science, statistics, or electrical engineering, and the Senior ML Engineer role involves work that directly relates to that field. The employer must also be enrolled in E-Verify. Your DSO confirms eligibility by matching your degree's CIP code to the official STEM designated degree list before recommending the extension on your I-20.
How do I confirm my employer is enrolled in E-Verify?
Use the E-Verify employer search tool to look up your prospective employer by company name or federal contractor status before signing an offer. E-Verify enrollment is a hard requirement for STEM OPT, not a preference. If the company isn't enrolled, they must complete enrollment before your STEM OPT start date, which adds processing time you'll want to account for before your initial OPT expires.
What goes in the I-983 training plan for a Senior ML Engineer?
Your I-983 must describe how the Senior ML Engineer role provides practical training in a STEM field directly related to your degree. Specific deliverables work better than job descriptions: include model architecture projects, data pipeline ownership, or production deployment responsibilities. Both you and a company supervisor must sign it, and your employer must report your training progress to your DSO every six months throughout the STEM OPT period.
What happens to my STEM OPT authorization if my employer loses E-Verify enrollment?
If your employer's E-Verify participation is terminated after your STEM OPT begins, your work authorization is no longer valid for that employer and you must stop working. USCIS expects continuous E-Verify enrollment throughout the STEM OPT period. Your DSO should be notified immediately, and you'll need to find a new E-Verify-enrolled employer or pursue a change of status if you have another option available.
How do I find Senior ML Engineer jobs that support STEM OPT?
Migrate Mate lists Senior ML Engineer roles filtered for employers enrolled in E-Verify and with documented sponsorship history, which removes the guesswork of identifying compliant companies. Because STEM OPT requires both E-Verify enrollment and a qualifying STEM role tied to your degree, starting with employers already familiar with the process shortens the time between offer and your training plan being signed.