AI Software Engineer

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 aStaff AI Software Engineerati Manage 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 willwork within the Applied AI team,collaborating with engineers,data scientists, and product stakeholders tobuildthe AI capabilities that poweri

Manage’senterprise work platform including Aski Manage, 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 whoidentifiesproblems before being asked, coordinates across teams, and drives solutions to completion. You think carefully about cost, performance, andreliabilityat scaleas well as communicate with clarity and accountability at every step.

• 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,genAIdocument intelligence, and agentic system use cases
• Collaborating with product and business stakeholders to translate requirements intoviabletechnical 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

• A Bachelor's, Master'sor 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
• Deepproficiencyin 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 moderngen
  
  AI/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) tooptimizesystems for scalability, latency, performance, operational complexity, and cost efficiency

• Experience in knowledge graph and multimodal LLMs
• Experience with distributed training (i.e,Pytorchdistributed, Ray) and inference optimization framework (i.e,vLLM,SGLang)
• Experience with agentic engineering (agent harness, agent memory) and orchestration framework like
  
  LangChain/Llama Index, 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;

s…