remote
Staff AI Software Engineer - iManage
Software Engineer
Lead end‑to‑end AI product development, from model research and fine‑tuning to production deployment and scaling, using Python, deep‑learning frameworks, and cloud‑native infrastructure.
About the role
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.
Skills
pythonpytorchtensorflowkubernetesawsdockermlflow