Judi Health is looking for a Machine Learning Engineer to help infuse advanced AI/ML capabilities into pharmacy benefits. You will join a mission-oriented startup focused on transforming healthcare through increased productivity and innovative user experiences. In this role, you will design and implement complex AI systems that leverage machine learning models for NLP, NLG, multimodal data analysis, chatbots, and RAG-based QnA.

What You'll Do

Develop and productionize machine learning solutions for Document Understanding, Search and QnA, GenAI, and Virtual Agents.
Develop and maintain backend services using Python, focusing on AI-driven applications.
Design and implement APIs for seamless integration with AI models and services.
Build tools for large-scale data processing and ETL, extracting insights to guide ML systems development.
Participate in code reviews, testing, and quality assurance processes.
Troubleshoot and resolve technical issues related to AI model integration, deployment, and backend services.
Develop algorithms to ensure the integrity and robustness of ML solutions through automated testing and validation.
Document and communicate development processes and implementation details with peers and supervisors.
Ensure the security and compliance of healthcare data, adhering to HIPAA regulations.

What We're Looking For

Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related quantitative field.

Nice to Have

2+ years of industry experience with a strong focus on ML solutions development and production deployment.
Good understanding of Machine Learning fundamentals such as Deep Neural Networks, Transformer-based LLM and MLMs, Boosted or standard decision trees, and RAG.

Technical Stack

Python
Large Language Models (LLM)
Multimodal Language Models (MLM)

Team & Environment

You will join a collaborative team within a mission-oriented, high-growth startup. We are excited to have team members who are passionate about making a lasting impact on transforming healthcare.