Staff AI Software Engineer - iManage

Key Responsibilities

• Design, develop, and ship AI models that solve complex business problems, overseeing the full lifecycle from data preprocessing to model serving.
• Build and maintain scalable, containerized inference pipelines on Kubernetes and AWS, ensuring high availability and low latency.
• Collaborate with data scientists, product managers, and engineering teams to translate research prototypes into production‑ready services.
• Implement CI/CD for machine‑learning workflows using tools such as MLflow, Docker, and GitOps practices.
• Monitor model performance in production, develop automated retraining and drift detection mechanisms, and continuously improve model accuracy and efficiency.

Requirements

• 5+ years of professional software engineering experience with a strong focus on machine learning and AI systems.
• Expertise in Python and deep‑learning libraries such as PyTorch or TensorFlow.
• Hands‑on experience deploying and scaling models on cloud platforms (AWS) using Kubernetes, Docker, and related DevOps tooling.
• Proven ability to design robust, production‑grade ML pipelines, including data versioning, model tracking, and automated testing.
• Strong problem‑solving skills, ability to work cross‑functionally, and a passion for mentoring junior engineers.