Senior Machine Learning Engineer I

Transformcap; United States; Hybrid; USD 140,000 - 190,000; Full time

Job summary

Judi Health is seeking a Senior Machine Learning Engineer to advance AI capabilities in pharmacy benefits, applying NLP, GenAI, and RAG techniques. You will design scalable ML systems, build backend services in Python, and implement multi-agent orchestration while ensuring secure handling of healthcare data.

The role emphasizes collaboration across product and engineering in a fast-paced startup, with a focus on deploying high-performance AI models in production and measuring impact with robust

Qualifications

• BS/MS in CS or related quantitative field.
• 4+ years of industry experience in ML solutions and deployment.
• Strong Python and PyTorch skills; experience with NLP/GenAI.
• Familiarity with cloud platforms (AWS/Azure) and containerized deployments.
• Experience designing evaluation frameworks for non-deterministic AI systems.

Responsibilities

• Design, develop, and productionize ML solutions for NLP, QnA, GenAI, and agents.
• Build backend services in Python for AI-driven applications.
• Create APIs for AI model integration and orchestration patterns.
• Scale RAG architectures; manage vector search and embedding pipelines.
• Collaborate with product and engineering across rapid, startup timelines.
• Lead ML lifecycle from problem framing to deployment and monitoring.
• Implement telemetry, A/B testing, and automated validation.
• Ensure HIPAA-compliant security and data governance.

Skills

Machine Learning

Python

NLP

GenAI

LLMs

RAG

Multimodal AI

Code review

Collaborative teamwork

Education

Bachelor’s or Master’s in CS/ML

Tools

PyTorch

AWS

Azure

Containerization

GenAI services

Job description

About Judi Health

Judi Health is a health technology company providing benefit administration solutions to employers, unions, health plans, and government entities. Judi Health replaces fragmented, outdated systems with the industry's first Unified Claims Processing architecture, seamlessly consolidating pharmacy and medical benefit administration on a single, secure platform. By delivering true price transparency, eliminating unnecessary middleman fees, and leveraging advanced AI-powered care delivery, Judi Health helps clients achieve unprecedented operational efficiency and service levels.
At Judi Health, we're deploying the infrastructure our country needs to deliver the healthcare we all deserve. We are the intelligence platform powering benefits plans for millions of Americans and proudly leading the next generation of care. To learn more, visit www.judi.health.

Location: Hybrid 3 days (offices in NYC, Denver, CO and Charlotte, NC area)

Position Summary:

Join our mission to infuse cutting-edge AI/ML/GenAI into pharmacy benefits as a Senior Machine Learning Engineer. We are looking for an experienced software engineer with machine learning expertise to join us in expanding our AI capabilities, enabling increased productivity and magical experiences in our products and services.

In this role you will be expected to design and implement complex AI systems that leverage ML models for NLP, NLG, multimodal data analysis, chatbots, and RAG-based QnA. The ideal candidate should be passionate about applying AI/ML concepts to difficult problems and develop scalable solutions. We want people who like working in a collaborative team environment and enjoy creating practical, efficient, and high-performance software that leverages Large Language Models (LLM), Multimodal Language Models(MLM), and other ML models and techniques to build amazing capabilities for our customers, partners, and employees. Most importantly, we are a mission-oriented, high-growth startup and we are looking for folks that are excited to be part of our journey to make lasting impact towards transforming healthcare.

Responsibilities:

• 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.
• Design multi-agent orchestration patterns (e.g. supervisor/subagent architectures, tool-use boundaries, human-in-the-loop gating) and evaluate architectural tradeoffs for scaling agent development across teams.
• Contribute to the development and scaling of the retrieval/search infrastructure underlying RAG systems (e.g. vector search clusters, embedding pipelines, capacity planning) - Build and maintain CI/CD pipelines supporting the build, test, and deployment of agentic AI systems.
• 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.

Required Qualifications:

• Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related quantitative field.
• Minimum 4+ 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, including hands-on experience with containerized deployments
• Familiarity with GenAI services such as OpenAI GPT, Anthropic Claude, Google Gemma etc. Experience developing and fine-tuning prompts on any of the GenAI services
• Experience designing evaluation frameworks for non-deterministic/agentic AI systems
• 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.

Salary Range

• New York, NY: $161,600 — $202,000 USD
• Denver, CO: $148,400 — $185,500 USD
• Charlotte, NC: $134,800 — $168,500 USD