Machine Learning Engineer

Judi Health • New York, NY

1d

About The Position

Join our mission to infuse cutting-edge AI/ML/GenAI into pharmacy benefits as a Machine Learning Engineer. We are looking for a 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.

Requirements

• Bachelor’s or Master’s degree in Computer Science, Machine Learning, or a related quantitative field.
• 2+ years experience in the industry with a strong focus on ML solutions development and production deployment is a plus.
• Good understanding of Machine Learning fundamentals such as Deep Neural Networks, Transformer-based LLM and MLMs, Boosted or standard decision trees, RAG, etc.
• Strong grasp of OOP, Design Patterns, efficient algorithms, and quality software development.
• Proficiency in Python and familiarity with ML libraries such as PyTorch.
• Familiarity with GenAI services such as OpenAI GPT, Anthropic Claude, Google Gemma etc. Some experience in developing and fine-tuning prompts on any of the GenAI services is a plus.
• Excellent problem-solving skills, attention to detail, and a strong capacity for logical thinking.
• The ability to work collaboratively across multiple disciplines in an extremely fast-paced, startup environment.
• Good written communication skills that enable collaboration in a remote environment.

Responsibilities

• 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.
• Develop tools for large-scale data processing and ETL and contribute to extracting insights from data to help 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 by developing automated testing and validation processes.
• 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 This Job Offers

• Job Type: Full-time
• Career Level: Mid Level