Staff AI Software Engineer

· Remote

Location

Remote

Type

Full Time

Job Description

Staff AI Software Engineer

Posted Yesterday
Be an Early Applicant
Chicago IL USA
Hybrid
180K-220K Annually
Senior level
Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
Whoever you are whatever you do however you work. Make it mean something at iManage.
The Role
The Staff AI Software Engineer will own the full AI systems lifecycle from model development to production deployment focusing on NLP and generative AI solutions while collaborating with cross-functional teams to optimize and drive solutions.
Summary Generated by Built In

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. 

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. 

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 

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+ year 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

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 

Don't meet every qualification listed above? Studies show that women and people of color are less likely to apply to jobs unless they meet all qualifications. At iManage we are committed to building a diverse and inclusive environment and encourage everyone to show up as their full authentic selves. We welcome those that come with a growth mindset and a hunger for learning; so if you are excited about this role but your past experience doesn't align perfectly with every qualification we encourage you to apply anyways!  

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.  
The overall US annual base salary range for this position is $180000–$220000. 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.  

Learn more at: www.imanage.com   

Please see our privacy statement for more information on how we handle your personal data:  

https://imanage.com/privacy-policy/   

#LI-Hybrid

#LI-RW1

What the Team is Saying

Jane
Bayardo
Lindsey
Suyash
Sandhya
Sam
Grace
Mark
Brian
Thomas
Jing
Stew
Nick
Neil

iManage Compensation & Benefits Highlights

  • Leave & Time Off BreadthPolicies highlight flexible non‑accrual PTO (iMAway) alongside paid sick leave and holidays. Some materials also reference wellness or volunteer time enabling broad time‑off options.
  • Parental & Family SupportParental leave is described as generous for both primary and secondary caregivers with inclusive gender‑agnostic language. Updates also note family medical leave childcare support and sensitivity to pregnancy loss.
  • Healthcare StrengthComprehensive coverage includes medical dental vision life and disability insurance with added services like HealthJoy and mental‑health tools (EAP Rethink Care/HealthyMinds). Wellness programs fitness initiatives and onsite/wellness rooms reinforce overall health support.

iManage Insights

Am I A Good Fit?
beta
Expert contributor network
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Chicago IL
1100 Employees
Year Founded: 2015

What We Do

iManage is dedicated to Making Knowledge Work. Over one million professionals across 65+ countries rely on our intelligent cloud-enabled secure knowledge work platform to uncover & activate the knowledge that exists inside their business content and communications. We are continuously innovating to solve the most complex professional challenges & enable better business outcomes. Our work is not always easy but it is ambitious & rewarding. We are a people-first company where our employee's wellbeing is our #1 priority! That’s why we offer generous paid time off private benefit packages that include Health/Vision/Dental/Life plus our employee assistance program (EAP) and HealthyMinds app to support your mental health and wellbeing. We realize that work is more than just a paycheck; it’s an important part of living and growing. We strive to have a positive impact on families and a commitment to creating an organization where everyone can thrive. We’re on a mission to foster an environment where everyone makes progress innovates and comes together as part of a best-in-class workplace. Everyone at iManage plays an important part in achieving our vivid future.

Why Work With Us

We’re looking for people who love a challenge. People who are happiest when they’re working hard solving problems and collaborating with the industry’s best and brightest. In exchange we’ll make sure the work you do here is worth doing. That’s the iManage way. It’s how we do things that might appear impossible.

Gallery

iManage Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

We provide flexibility for our employees to have a meaningful work-life balance as remote workers & an option to work from 1 of 7 global offices

Typical time on-site: 0 days a week
HQChicago IL
Sydney
Belfast GB
Bengaluru IN
London GB
San Francisco CA
Sunnyvale California
Toronto Ontario
Learn more

Similar Jobs

iManage

Senior Customer Success Manager

Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
Hybrid
Chicago IL USA
1100 Employees
105K-155K Annually

iManage

Account Manager

Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
Remote or Hybrid
United States
1100 Employees
120K-150K Annually

iManage

Head of Field Marketing

Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
Hybrid
Chicago IL USA
1100 Employees
180K-210K Annually

iManage

Development Manager

Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
Hybrid
Chicago IL USA
1100 Employees
145K-195K Annually
Apply Now

Date Posted

05/07/2026

Views

0

Back to Job Listings Add To Job List Company Profile View Company Reviews
Neutral
Subjectivity Score: 0
142,000+ Jobs Tracked
12,400+ Companies
1,930 Categories