Principal Machine Learning Engineer

Summary

Join SandboxAQ's AQMed team as a Principal Machine Learning Engineer to contribute to groundbreaking cardiac diagnostics. You will design, train, and deploy state-of-the-art deep learning models using diverse data sources. This role requires extensive experience in deep learning, biomedical modeling, and working with large datasets. You will collaborate with a world-class team, external partners, and stakeholders to translate research into impactful clinical applications. The position offers a competitive salary, bonus potential, equity, and a comprehensive benefits package. This is an opportunity to make a significant contribution to medical innovation within a dynamic and supportive environment.

Requirements

• 10+ years as a data scientist or MLE, including experience working with time series data, biomedical models (especially cardiac), and training AI models on very large datasets
• Masters or PhD in computer science, data science, mathematics, physics, AI or a related field
• Developed and deployed state-of-the-art deep learning models to production for real-world applications, especially in MedTech
• Shipped products, bringing practical, seasoned knowledge to navigate challenges and deliver results
• Proficient in advanced DL/ML frameworks, with a focus on self-supervised methods like contrastive learning, generative modeling, masked modeling, graph networks, and clustering
• Skilled in hyperparameter tuning, explainability, and transfer learning
• Proficient in Python and using versioning control systems (e.g., git), show high-quality code standards, and actively collaborate with other scientists and researchers reviewing PRs and writing scalable, maintainable code
• Strong foundation in statistics, probability, and data wrangling from varied sources
• Comfortable with AWS, docker, and scalable batching pipelines
• Skilled at bridging technical, clinical, and business perspectives by presenting insights tailored to diverse stakeholders
• Builds intuitive dashboards and visualization tools to drive shared understanding, alignment, and actionable decisions across teams
• Thrive in ambiguity, take ownership, balance competing priorities effectively, and excel at translating a compelling vision into technical reality

Responsibilities

• Hone your technical leadership by applying hard-won practical experience in deep learning to tackle challenging yet rewarding problems at the forefront of medical science, while collaborating with a world-class team
• Design, train, and fine-tune state-of-the-art deep learning architectures that seamlessly integrate multimodal data for robust and scalable solutions
• Harness cutting-edge AI techniques to uncover novel biomarkers, combining insights from the latest research, experimental data, and explainability frameworks to drive groundbreaking clinical applications
• Contribute to building and refining pipelines that power a sophisticated custom AI stack, delivering precise predictions, actionable insights, and user-friendly visualizations tailored for medical and scientific innovation
• Develop tools for explainable AI and integrate findings into production systems
• Build dashboards and tools to communicate progress and performance using AWS and other 3rd party tools
• Adeptly present insights to cross-functional teams as well as internal and external stakeholders, targeting your communication to the particular audience and anticipating the questions and proof points that speak best to their perspective
• Mentor junior team members, fostering a collaborative and innovative team environment with people management potential in the future

Preferred Qualifications

• Experience training large models on GPUs
• Working with LLMs
• Using synthetic data to augment small real-world datasets

Benefits

• Competitive salaries
• Stock options depending on employment type
• Generous learning opportunities
• Medical/dental/vision
• Family planning/fertility
• PTO (summer and winter breaks)
• Financial wellness resources
• 401(k) plans
• Annual discretionary bonuses
• Equity