Senior Machine Learning Engineer

About the role

We are looking for a highly skilled Senior Machine Learning Engineer to lead the evolution of our core Python SDK frameworks and Agentic AI capabilities.

Key Responsibilities

• Lead the design and implementation of Feature Enhancements for internal Python SDK frameworks to expand functionality and improve developer experience.
• Provide expert-level technical support and proactive maintenance for the Python SDK ecosystem, ensuring stability, reliability, and seamless adoption across the organization.
• Develop and scale agentic workflows using modern Agentic Frameworks to execute intricate business processes with minimal oversight.
• Architect and deploy scalable, high-performance APIs using FastAPI to serve as the backbone for machine learning services and agent interactions.
• Design and maintain robust automated pipelines using Jenkins and GitLab Runners to ensure continuous integration, testing, and deployment of ML tools.
• Utilize Galileo for model evaluation and observability, ensuring that agentic workflows remain accurate, safe, and performant in production environments.

Basic Qualifications

• Mastery of Python and its machine learning ecosystem, with extensive experience building production-ready services via FastAPI.
• Proficiency in building and scaling agentic workflows and multi-agent systems.
• Hands-on experience with Jenkins and GitLab Runners for automating software lifecycles and managing SDK distribution.
• Experience working with standardized application templates (e.g., PyVegas) and evaluation platforms like Galileo.
• Proven ability to design scalable infrastructure and maintain complex software frameworks used by other engineering teams.
• Strong analytical skills to debug complex SDK interactions and logic loops.
• Ability to collaborate with cross-functional teams to define requirements for autonomous systems and developer tools.
• Proven ability to drive high-impact projects from research to production with minimal supervision.

Good to Have

• Contributions to open-source Python SDKs or Agentic AI projects.
• Experience with MLOps practices for the continuous deployment of agentic systems.
• Familiarity with containerization (Docker/Kubernetes) for deploying FastAPI-based microservices.