OPT AI ML Platform Jobs
AI ML Platform engineering sits at the intersection of infrastructure and machine learning, making it one of the most OPT-friendly specializations in tech. Employers in this space regularly sponsor H-1B visas, and STEM OPT extensions give you up to three years of work authorization to build the track record that makes sponsorship straightforward.
Find OPT AI ML Platform JobsOverview
Showing 5 of 299+ AI ML Platform jobs










See all 299+ AI ML Platform Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new AI ML Platform roles.
Get Access To All Jobs
DESCRIPTION
Do you want to join an innovative team of scientists who develop Agentic AI, LLM, and deep learning based solutions to help Amazon provide the best seller experience across the entire Seller life cycle, including recruitment, growth, support, risk mitigation and provide the best customer and seller experience?
Do you want to build advanced algorithmic systems that help manage the trust and safety of millions of customer interactions every day? Are you excited by the prospect of analyzing and modeling terabytes of data and creating state-of-the-art algorithms to solve real world problems? Are you excited by the opportunity to leverage GenAI and innovate on top of the state-of-the-art large language models to improve customer and seller experience?
Do you like to build end-to-end business solutions and directly impact the profitability of the company? Do you like to innovate and create solutions that have cross-organizational impacts?
If yes, then you may be a great fit to join the Machine Learning Accelerator team.
Key job responsibilities
The scope of an Applied Scientist in the Machine Learning Accelerator (MLA) team is to research and prototype AI and Machine Learning applications that solve strategic business problems across Selling Partner Experience (SPX) domains. Additionally, the scientist collaborates with engineers and business partners to design and implement solutions at scale that are of broad benefit to SPX organizations. They develop large-scale solutions for high impact projects, introduce tools and other techniques that can be used to solve problems from various perspectives, and show depth and competence in more than one area. They influence the team’s technical strategy by making insightful contributions to the team’s priorities, approach and planning. They develop and introduce tools and practices that streamline the work of the team, and they mentor junior team members and participate in hiring.
BASIC QUALIFICATIONS
- PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience
- 3+ years of building machine learning models or developing algorithms for business application experience
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- Experience programming in Java, C++, Python or related language
- Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing
- Experience with Machine Learning and Large Language Model fundamentals, including architecture, training/inference lifecycles, and optimization of model execution, or experience in developing and deploying LLMs in production on GPUs, Neuron, TPU or other AI acceleration hardware
PREFERRED QUALIFICATIONS
- Experience using Unix/Linux
- Experience in state-of-the-art deep learning models architecture design and deep learning training and optimization and model pruning
- Experience with LLM fine-tuning, in-context learning, or model evaluation
- Hands-on experience designing reward models or RL post-training pipelines (PPO/GRPO, DPO) for LLMs or agents, including preference-data collection and evaluation
- Experience building agentic AI systems — tool use, planning, retrieval-augmented generation, or multi-agent workflows
- Experience with researching and developing neuro-symbolic solutions and applications
- Publications at top ML/AI venues
- Experience partnering with product/engineering teams to deliver ML in large-scale production system
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.
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, CA, San Diego
COMPENSATION
- Salary Range: $142,800.00 - $193,200.00 USD annually
See all 299+ OPT AI ML Platform Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new OPT AI ML Platform Jobs.
Get Access To All JobsTips for Finding OPT Sponsorship in AI ML Platform
Target companies with dedicated ML infrastructure teams
Large tech firms and AI-focused startups with dedicated ML platform teams sponsor OPT and H-1B at high rates. Look for engineering organizations where ML infrastructure is a core product function, not a side project maintained by a single engineer.
Lead with platform impact, not just model work
Hiring managers for ML platform roles want engineers who reduce friction for data scientists at scale. Frame your experience around compute cost reduction, training throughput improvements, or pipeline reliability, not just model accuracy benchmarks from academic projects.
Get your STEM OPT extension filed before your OPT expires
Your STEM OPT extension application must be filed at least 90 days before your initial OPT end date. Missing this window means a gap in work authorization. Coordinate early with your DSO and confirm your employer qualifies under E-Verify requirements.
Specialize in a high-demand ML platform layer
Feature stores, model registries, distributed training orchestration, and inference serving are all understaffed relative to demand. Deep expertise in one of these layers makes you a stronger H-1B petition candidate because the specialty occupation requirement is easier to document.
Understand your employer's H-1B sponsorship timeline
H-1B cap-subject petitions can only be filed in April for an October start. If your OPT expires before October, ask whether the employer will file for cap-exempt status or use an academic or nonprofit partner. Clarify this before accepting an offer.
Document your work authorization status clearly in applications
When applying, state your OPT status, your STEM extension eligibility, and your authorization end date upfront. Employers unfamiliar with OPT often assume it means immediate sponsorship costs. Showing you have one to three years of self-authorized work time removes that friction.
AI ML Platform OPT: Frequently Asked Questions
Do AI ML Platform jobs typically qualify for the STEM OPT extension?
Yes. AI ML Platform roles are almost always classified under CIP codes in computer science or electrical engineering, both of which appear on the STEM Designated Degree Program list. As long as your degree is in a qualifying STEM field and your employer participates in E-Verify, you can extend your OPT by 24 months beyond the initial 12, giving you up to three years of work authorization total.
How do I find AI ML Platform jobs that are open to OPT students?
Migrate Mate is built specifically for F-1 OPT students and filters job listings by sponsorship willingness, so you're not wasting time applying to roles where OPT is a disqualifier. AI ML Platform roles are well-represented on the platform because the employers posting there are already accustomed to hiring international talent on OPT and H-1B.
Can I work on an AI ML Platform team at a startup on OPT?
Yes, but verify the startup participates in E-Verify before accepting an offer if you plan to apply for a STEM OPT extension. E-Verify enrollment is a federal requirement for STEM OPT employers, and smaller startups sometimes haven't completed it. Enrollment takes only a few days, so most employers are willing to complete it if they want to hire you.
What makes AI ML Platform a strong field for long-term visa sponsorship?
ML platform engineering is a well-defined specialty occupation with a clear degree requirement in computer science or a related engineering field, which makes H-1B petitions more straightforward to document than generalist software roles. Demand consistently exceeds supply, so employers in this space have strong financial incentives to sponsor rather than lose a qualified candidate to a competitor.
What should I do if my OPT expires before my employer's H-1B petition is approved?
If you filed your H-1B petition before your OPT expired and your status is maintained through cap-gap coverage, you may continue working while the petition is pending. Cap-gap automatically bridges the period between OPT expiration and the October 1 H-1B start date for timely-filed petitions. Confirm with your DSO that your I-20 reflects the cap-gap extension and that your employer's attorney filed before the April deadline.