iManage Logo

iManage

Staff AI Software Engineer

Posted 2 Days Ago
Be an Early Applicant
Hybrid
Toronto, ON
Mid level
Hybrid
Toronto, ON
Mid level
Lead end-to-end ML lifecycle for enterprise NLP and generative AI systems: prototype, fine-tune, deploy, and scale LLM/NLP solutions on GPU-backed Kubernetes cloud infrastructure while ensuring observability, CI/CD, performance, and cost efficiency.
The summary above was generated by AI

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 & Fridays 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 rapidly evolving, industry-leading SaaS company on an exciting journey of growth and scalability!
  • Take on meaningful, high-impact challenges by leveraging cutting-edge technologies and best-in-class protocols to drive innovation.
  • Own my career path with our internal development framework. Ask us more about this!
  • Expand my skill set and earn certifications with unlimited access to LinkedIn Learning courses and interactive Microsoft courses & training.
  • Be part of a supportive and experienced team within a dynamic, inclusive, and encouraging culture.
  • Enjoy flexible work hours that empower me to balance personal time with professional commitments.
  • Collaborate in a modern, open-plan workspace featuring a gaming area, free snacks and drinks, and regular social events.

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 Registered Retirement Savings Plan with a company match up to 5%.
  • 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.
  • Having multiple company wellness days each year to prioritize mental health and well-being.
  • Providing access to RethinkCare, a global behavioral health platform that enhances personal well-being, strengthens professional resilience, and empowers parental success through expert-led training and resources.

iManage is committed to providing an excellent candidate experience and will never ask you to engage in recruitment activity via text and exclusively communicates from emails using the @imanage.com domain. If you have any concerns or questions about communications you have received, please send them to [email protected] so our team members can review.

About iManage…

At iManage, we are dedicated to Making Knowledge Work™. Our intelligent, cloud-enabled, and secure platform is trusted by 4,100+ customers and 430,000 users worldwide, managing over 11 billion documents and 11 petabytes of data. We empower professionals across 65+ countries to unlock the full potential of 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 embrace challenges. People who thrive on solving problems, pushing boundaries, and collaborating with the industry’s best and brightest. That’s the iManage way. It’s how we turn the impossible into reality, empower our employees to grow, unlock their potential, and create a meaningful impact on 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 recruitment privacy statement for more information on how we handle your personal data.

#LI-Hybrid

#LI-RW1

Similar Jobs at iManage

10 Days Ago
Hybrid
Senior level
Senior level
Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
The Senior Site Reliability Engineer at iManage will build and maintain resilient platforms, drive reliability best practices, and support cloud infrastructure with automation and scalable solutions.
Top Skills: AksAzureBashChefDockerEfkElkGoGrafanaJavaKubernetesLinuxPowershellPrometheusPythonRubyTerraform
15 Days Ago
Hybrid
Mid level
Mid level
Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
As a Site Reliability Engineer for Data Services at iManage, you will ensure data durability, optimize performance, manage upgrades, and collaborate across teams, focusing on reliability and sustainability in cloud-native environments.
Top Skills: AzureBashDockerEfkElasticsearchElkGoGrafanaHashicorp TerraformJavaKubernetesMariadbMaxscalePowershellPrometheusPythonRuby
15 Days Ago
Hybrid
Mid level
Mid level
Artificial Intelligence • Cloud • Information Technology • Legal Tech • Productivity • Software
As a Site Reliability Engineer, you will create middleware, automate processes, maintain cloud infrastructure, and participate in on-call rotations while driving improvements in platform security and resilience.
Top Skills: AksAzureBashChefDockerEfkElkGoGrafanaJavaKubernetesPowershellPrometheusPythonRubyTerraform

What you need to know about the Vancouver Tech Scene

Raincouver, Vancity, The Big Smoke — Vancouver is known by many names, and in recent years, it has gained a reputation as a growing hub for both tech and sustainability. Renowned for its natural beauty, the city has become a magnet for professionals eager to create environmental solutions, and with an emphasis on clean technology, renewable energy and environmental innovation, it's attracted companies across various industries, all working toward a shared goal: advancing clean technology.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account