Senior Machine Learning Engineer II

• Design, develop, and productionize machine learning (ML) solutions in the fields of Document understanding, Search and QnA, GenAI, Virtual Agents, etc.
• Develop and maintain backend services using Python, focusing on AI-driven applications.
• Design and implement APIs for seamless integration with AI models and services.
• Collaborate with product managers, engineers, and other stakeholders as a specialist and subject matter expert in machine learning.
• Work across the complete lifecycle of ML model development, including problem definition, data exploration, feature engineering, model training, validation, and deployment.
• Participate in code reviews, testing, and quality assurance processes.
• Troubleshoot and resolve technical issues related to AI model, integration, deployment and backend services.
• Implement system telemetry, A/B testing and other statistical methods to validate the effectiveness of AI models. Ensure the integrity and robustness of ML solutions by developing automated testing and validation processes.
• Document and communicate development processes, system designs, and architectural decisions.
• Ensure the security and compliance of healthcare data, adhering to HIPAA regulations.

• Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related quantitative field.
• Minimum 6+ years of experience in industry with a strong focus on ML solutions development and production deployment.
• Strong grasp of OOP, Design Patterns, efficient algorithms, and quality software development.
• Strong proficiency in Python and familiarity with ML libraries such as PyTorch.
• Statistical Machine Learning and Artificial Intelligence in the context of NLP, Speech, RAG, Virtual Agents, and/or Document Processing technologies
• Familiarity with cloud platforms such as AWS or Azure.
• Familiarity with GenAI services such as OpenAI GPT, Anthropic Claude, Google Gemma etc.

developing and fine-tuning prompts on any of the GenAI services

• Excellent problem-solving skills, attention to detail, and a strong capacity for logical thinking.
• Passion for customer, innovation, product quality, service excellence, and test engineering
• The ability to work collaboratively across multiple disciplines in an extremely fast-paced, startup environment.
• Experience working with microservices/an API-based application paradigm.
• Good written communication skills that enable collaboration in a remote environment.

This position description is designed to be flexible, allowing management the opportunity to assign or reassign duties and responsibilities as needed to best meet organizational goals.

All employees are responsible for adherence to the Judi Health Code of Conduct including the reporting of non-compliance.

This position description is designed to be flexible, allowing management the opportunity to assign or reassign duties and responsibilities as needed to best meet organizational goals.

We provide equal employment opportunities to all employees and applicants for employment and prohibit discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, medical condition, genetic information, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.

By submitting an application, you agree to the retention of your personal data for consideration for a future position at Judi Health.

More details about Judi Health's privacy practices can be found at https://www.judi.health/legal/privacy-policy .