STEM OPT AI Software Developer Jobs
AI Software Developer roles in machine learning, NLP, and applied AI fall squarely within STEM OPT eligibility. Your STEM degree qualifies you for a 24-month OPT extension beyond the initial 12 months, giving you up to 36 months of work authorization. Employers must be enrolled in E-Verify to hire you under STEM OPT.
Find STEM OPT AI Software Developer JobsOverview
Showing 5 of 305+ AI Software Developer jobs


Have you applied for this role?


Have you applied for this role?


Have you applied for this role?


Have you applied for this role?


Have you applied for this role?
See all 305+ AI Software Developer Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new AI Software Developer roles.
Get Access To All Jobs
INTRODUCTION
We offer a flexible working policy that supports a healthy balance between personal and professional well-being. This role requires in-office presence on Tuesdays & Thursdays to collaborate, connect, and learn from peers - while also maintaining the flexibility for meaningful work-life balance.
ABOUT THE ROLE
Being a Staff AI Software Engineer at iManage Means…
You are passionate about owning the full lifecycle of AI systems, from training and fine-tuning models to deploying and scaling them in production. You thrive at the intersection of machine learning, data science, and infrastructure engineering, and you bring both analytical accuracy and hands-on technical depth to everything you build. You will work within the Applied AI team, collaborating with engineers, data scientists, and product stakeholders to build the AI capabilities that power iManage’s enterprise work platform including Ask iManage, our generative AI document assistant, as well as advanced document classification, extraction, and NLP-driven features that directly impact how knowledge workers search, organize, and understand their most critical information. You will tackle some of the most interesting problems in applying large language models and NLP at scale to enterprise documents and workflows.
You are a self-starter who identifies problems before being asked, coordinates across teams, and drives solutions to completion. You think carefully about cost, performance, and reliability at scale as well as communicate with clarity and accountability at every step.
Responsibilities
iM Responsible For...
- Owning the end-to-end ML lifecycle for AI systems from prototype to production, from model development and evaluation to scalable application design and production serving for NLP, genAI document intelligence, and agentic system use cases
- Collaborating with product and business stakeholders to translate requirements into viable technical solutions
- Deploying and optimizing ML/AI systems on GPUs and Kubernetes-based cloud infrastructure, including AKS or equivalent platforms, while balancing trade-offs across scalability, latency, performance, operational complexity, and cost efficiency
- Applying modern engineering practices for production AI systems, including monitoring, observability, integration testing, containerized services, CI/CD pipelines, model/version tracking, and release governance
- Driving architectural decisions for AI systems and mentoring team members to foster a culture of innovation and knowledge sharing
BASIC QUALIFICATIONS
iM Qualified Because I Have…
- A Bachelor's, Master's or Ph.D. in Computer Science, Machine Learning, Data Science, Artificial Intelligence, Statistics, or a related field
- The ability to work collaboratively across teams, communicate with precision, and take ownership from prototype to production
- 4+ years of experience in ML/AI engineering or software engineering with 3+ years of experience building and shipping in NLP and LLM systems into productions
- Deep proficiency in Python and modern AI frameworks (i.e., PyTorch and Hugging Face)
- Deep understanding of ML fundamentals including hands-on experience with both traditional ML/NLP and modern genAI/transformer architectures
- Demonstrated hands-on experience in fine-tuning language models and deploying them to production
- Experience with GPU optimization for both training and inference
- Experience with k8s deployment on cloud infrastructure (Azure/AWS/GCP) to optimize systems for scalability, latency, performance, operational complexity, and cost efficiency
PREFERRED QUALIFICATIONS
Bonus Points If I Have...
- Experience in knowledge graph and multimodal LLMs
- Experience with distributed training (i.e., Pytorch distributed, Ray) and inference optimization framework (i.e., vLLM, SGLang)
- Experience with agentic engineering (agent harness, agent memory) and orchestration framework like LangChain/LlamaIndex, or similar tools
WHAT WE OFFER
iM Getting To…
- Join a supportive, experienced team with an inclusive, encouraging, and vibrant culture
- Have flexible work hours that allow me to balance my ‘me time’ with my work commitments
- Collaborate in a modern open plan workspace, with a gaming area, free snacks, drinks and regular social events
- Focus on impactful work, solving complex, real challenges utilizing the latest technologies and protocols
- Own my career path with our internal development framework. Ask us more about this!
- Learn new skills and earn certifications with access to unlimited courses in LinkedIn Learning
- Join an innovative, industry leading SaaS company that is continuing to grow & scale!
iManage Is Supporting Me By...
- Creating an inclusive environment where I can help shape the culture not just by fitting in, but by adding to it
- Providing a market competitive salary that is applied through a consistent process, equitable for all our employees, and regularly reviewed based on industry data
- Rewarding me with an annual performance-based bonus
- Offering comprehensive Health/Vision/Dental/Life Insurance, and a 401k Retirement Savings Plan with a company match up to 4%
- Giving access to HealthJoy, a healthcare concierge service, to help me maximize my health benefits
- Granting enhanced leave for expecting parents; 20 weeks 100% paid for primary leave, and 10 weeks 100% paid for secondary leave
- Providing me with a flexible time off policy to take the time off that I need. Be it for vacation, volunteering, celebrating holidays, spending time with family, or simply taking time to recharge and reset
- Caring for my mental health and well-being with multiple company wellness days and free access to the Healthy Minds app for mindfulness, meditation and more
COMPENSATION
The overall US annual base salary range for this position is $180,000–$220,000. Individual compensation for each candidate depends on factors such as qualifications, experience, and candidate location. This range does not include additional forms of compensation, such as bonuses, or benefits. Your recruiter will provide further details about the offer range, incentives, and overall compensation during the hiring process.
ABOUT iManage…
iManage is dedicated to Making Knowledge WorkTM. Over one million professionals across 65+ countries rely on our intelligent, cloud-enabled, secure knowledge work platform to uncover and activate the knowledge that exists inside their business content and communications.
We are continuously innovating to solve the most complex professional challenges and enable better business outcomes; our work is not always easy but it is ambitious and rewarding.
So we’re looking for people who love a challenge. People who are happiest when they’re solving problems and collaborating with the industry’s best and brightest. That’s the iManage way. It’s how we do things that might appear impossible. How we develop our employees’ strengths and unlock their potential. How we find meaning in everything we do.
Whoever you are, whatever you do, however you work. Make it mean something at iManage.
iManage provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.
This policy applies to all terms and conditions of employment, including recruiting, hiring, placement, promotion, termination, layoff, recall, transfer, leaves of absence, compensation and training.
See all 305+ STEM OPT AI Software Developer Jobs
Sign up for free to unlock all listings, filter by visa type, and get alerts for new STEM OPT AI Software Developer Jobs.
Get Access To All JobsTips for Finding STEM OPT Authorization as an AI Software Developer
Verify your CIP code matches AI roles
Check that your degree's Classification of Instructional Programs code falls under an approved STEM field before applying. Computer science, computer engineering, and data science CIP codes are typically eligible. Your DSO can confirm your specific code against the STEM OPT designated degree program list.
Confirm E-Verify enrollment before accepting offers
Ask hiring managers directly whether the company is currently enrolled in E-Verify, not just registered. Some startups let enrollment lapse. You can also search the E-Verify employer database yourself to verify active participation status before you invest time in their interview process.
Tailor your resume to AI job requirements
AI Software Developer postings frequently list specific frameworks like PyTorch, TensorFlow, or LangChain as hard requirements. Align your resume's skills section to the exact tools named in each job description so applicant tracking systems don't filter you out before a recruiter reviews your application.
Target employers with active AI LCA filings
Use Migrate Mate to filter for employers with verified Labor Condition Application filings in AI and software development roles. LCA history signals that the company understands STEM OPT compliance and has filed the required DOL documentation for international workers before.
File your I-983 training plan before your start date
Your employer must complete and sign the I-983 before you begin work on the STEM OPT extension. AI roles should map specific learning objectives, like model deployment or system architecture skills, to your degree field. A vague training plan increases the risk of a SEVIS compliance issue later.
Understand cap-gap protection during H-1B transition
If your employer files an H-1B visa petition before your STEM OPT expires and you're selected, cap-gap rules extend your work authorization through September 30. Make sure your employer files by the April 1 deadline so there's no gap between your STEM OPT end date and H-1B start.
Frequently Asked Questions
Does my degree qualify me for STEM OPT as an AI Software Developer?
Your degree qualifies if it appears on the STEM Designated Degree Program List maintained by the Department of Homeland Security. Degrees in computer science, computer engineering, electrical engineering, mathematics, and data science typically qualify. The classification is based on your degree's CIP code, not your job title. Your DSO can confirm eligibility using your official transcript and I-20.
Does my employer need to be enrolled in E-Verify to hire me on STEM OPT?
Yes. E-Verify enrollment is a hard requirement for STEM OPT, not optional. The employer must be actively enrolled at the time you begin work, not just registered in the past. You can verify a company's current enrollment status through the E-Verify employer search tool. If an employer isn't enrolled, you cannot work there on your STEM OPT extension regardless of the job offer.
What goes into the I-983 training plan for an AI Software Developer role?
The I-983 must connect your specific work activities to the STEM field your degree covers. For an AI Software Developer, this means documenting learning objectives tied to tasks like designing machine learning pipelines, building inference systems, or applying statistical modeling. Generic job descriptions don't satisfy USCIS requirements. Your employer signs the form, and your DSO must receive and validate it before your extension begins.
How does cap-gap work if my STEM OPT expires during H-1B processing?
If your employer submits a timely H-1B cap-subject petition before April 1 and you're selected in the lottery, cap-gap protection automatically extends your work authorization through September 30 of that year. This covers the gap between your STEM OPT expiration and the October 1 H-1B start date. Your employer must file on time for this protection to apply, so confirm their filing timeline well in advance.
Where can I find AI Software Developer jobs where employers already understand STEM OPT?
Migrate Mate filters AI Software Developer roles by employers with verified DOL Labor Condition Application filing history, which indicates they've hired international workers and understand the compliance requirements. Targeting these employers reduces the back-and-forth of explaining STEM OPT mechanics to HR teams unfamiliar with E-Verify enrollment or I-983 obligations.