Senior AI Software Engineer

Senior AI Software Engineer – iManage

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 and Fridays to collaborate, connect, and learn from peers while maintaining flexibility for meaningful work‑life balance.

• Owning the end‑to‑end ML lifecycle for AI systems from model development and evaluation through to scalable production serving, generative AI document intelligence, and agentic system use cases
• 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
• Designing and implementing production‑ready LLM applications and APIs with monitoring, observability, and integration testing built in
• Applying modern engineering practices for production AI systems, including containerized services, CI/CD pipelines, model and version tracking, and release governance
• Collaborating with product and business stakeholders to translate requirements into viable technical solutions
• Conducting code reviews and providing constructive feedback to team members
• Contributing to best practices and standards for AI engineering across the team

• A Bachelor’s, Master’s, or Ph.D. in Computer Science, Machine Learning, Data Science, Artificial Intelligence, Statistics, or a related field
• 3+ years of experience in ML/AI engineering or software engineering
• Hands‑on experience building and shipping LLM systems into production
• Deep proficiency in Python and modern AI frameworks, including PyTorch and Hugging Face
• Solid understanding of ML fundamentals, including hands‑on experience with both traditional ML and modern generative AI and transformer architectures
• Demonstrated experience fine‑tuning language models and deploying them to production
• Experience with GPU optimization for training and inference workloads
• Experience with Kubernetes deployment on cloud infrastructure (Azure, AWS, or GCP) to optimize systems for scalability, latency, performance, and cost efficiency
• The ability to work collaboratively across teams, communicate with precision, and take ownership from prototype to production

• Experience with distributed training frameworks (e.g., PyTorch Distributed, Ray) and inference optimization tools (e.g., vLLM, SGLang)
• Experience with agentic engineering, including agent harness and agent memory, and orchestration frameworks such as Lang Chain, Llama Index, or similar tools
• Experience with knowledge graphs and multimodal LLMs
• Familiarity with AI/ML observability tools and model lifecycle management best practices

• 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 Rethink Care, a global behavioral health platform that enhances personal well‑being, strengthens professional resilience, and empowers parental success through expert‑led training and resources

• 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 your career path with our internal development framework
• Expand your…