Applied Machine Learning Engineer

Summary

Join Lightbox Health, a cutting-edge healthcare technology company, as a Staff Applied Machine Learning Engineer and be a founding team member. Lead the development and optimization of ML systems, focusing on training models, building pipelines, and fine-tuning LLMs for healthcare applications. You will collaborate with cross-functional teams, mentor junior engineers, and communicate technical findings to stakeholders. This role requires extensive experience in machine learning and leadership, along with proficiency in relevant programming languages and frameworks. Lightbox offers a comprehensive benefits package including full health coverage, paid parental leave, unlimited PTO, professional development opportunities, and remote work options.

Requirements

• 10+ years of experience in computer science or an equivalent field
• Proven experience in a leadership role driving ML system development and optimization, preferably in healthcare or related fields
• Demonstrated expertise in training ML models and building robust training pipelines for real-world applications
• Strong understanding of machine learning frameworks such as TensorFlow, PyTorch, or similar
• Proficient in programming languages like Python or Go, with the ability to write efficient, clean, and maintainable code
• Excellent written and verbal communication skills, with the ability to convey technical concepts to both technical and non-technical audiences
• A track record of delivering impactful machine learning solutions that have been successfully deployed in production environments

Responsibilities

• Lead the team in architecting, building, and optimizing ML systems to deliver high-quality, real-world results in healthcare settings
• Design and implement robust training pipelines for machine learning models, ensuring efficiency and scalability for healthcare data
• Fine-tune ML models to meet specific healthcare needs and optimize their performance for various medical applications
• Develop and implement feedback mechanisms to continuously improve the accuracy and effectiveness of ML in healthcare contexts
• Collaborate with cross-functional teams to understand healthcare business requirements and translate them into actionable ML solutions
• Stay up-to-date with the latest advancements in machine learning and healthcare technology, implementing best practices to enhance our ML infrastructure
• Coach and mentor junior data engineers, fostering a culture of continuous learning and growth within the Lightbox Health team
• Communicate complex technical concepts and findings to non-technical stakeholders in a clear and concise manner, particularly in healthcare contexts

Preferred Qualifications

• Familiarity with healthcare data privacy regulations and best practices for handling sensitive information
• Expertise in NLP and document processing techniques such as OCR, entity extraction, or embedding generation, particularly for healthcare documents
• Experience with algorithms for searching, sorting, and retrieval of structured and unstructured data, with familiarity in tools like vector databases, graph databases, ElasticSearch, or similar
• Understanding of medical claims processing, payer-provider workflows, and healthcare-related standards like ICD, CPT, SOC2, FHIR
• Familiarity with MLOps best practices, including CI/CD, model versioning, monitoring, and retraining workflows
• Experience designing systems with a focus on low-latency APIs or data pipelines for high-performance environments

Benefits

• 100% employer paid, comprehensive health care including medical, dental, and vision for you and your family
• Paid maternity and paternity for 14 weeks at employees' normal pay
• Unlimited PTO, with management approval with all Federal holidays observed
• Opportunities for professional development and continued learning in healthcare technology
• Occasional travel to collaborate with your team in person
• Optional 401K and FSA available
• Ability to work fully remote within the United States