Sr. Machine Learning Engineer
Summary
Senior ML Engineer developing and deploying deep learning models to automate orthodontic treatment planning, focusing on 3D geometric processing and multimodal data fusion using Python, PyTorch, and AWS.
Align Technology is a global medical device company that is changing lives through better smiles. We revolutionized the orthodontic industry and the way orthodontic and restorative treatment is presented and delivered to millions of people around the world.
You will be part of a cross-functional and multiregional development organization utilizing a broad set of technologies and tools. As a Senior Machine Learning Engineer, you will develop, validate, and deploy features for internal applications aimed at automating the treatment planning process. The position emphasizes 3D geometric processing, multimodal data fusion, and the application of advanced machine learning techniques to complex clinical workflows.
In this role, you will…
As a Senior Machine Learning Engineer you will have following responsibilities:
- Decompose initial (poorly) stated business requests into separate technical problems and identify a specific ML toolset for a particular problem
- Apply machine learning approaches leading to improvements in orthodontic treatment planning
- Design and develop deep learning models for different data sources (e.g. 3D meshes, free-form text, table data etc.)
- Validate and analyse models performance
- Collaborate with the development team on architecture design of proposed solutions and support integration of the model into the production pipeline
- Document and transfer technology to product groups across the company
- Ownership of a particular ML component, its maintenance, and improvements
You will work with following technological stack:
- Python
- PyTorch
- Docker
- Amazon Cloud
- Git/Bitbucket
In this role, you’ll need …
We are looking for candidates with the following skills:
- Master's degree or higher in Computer Science, Statistics, Machine Learning, Statistical Data Modelling or related fields
- 8+ years of experience in deep learning applications for commercial projects (preferably in computer vision tasks)
- Solid understanding of deep learning approaches, their limitations and powers (e.g. for semantic/instance segmentation tasks, Visual Transformers, modern LLMs, fusions of different data types, etc.)
- Proficiency in the standard technological stack: Python, git, Linux, Docker
- Mathematical background for solving ML tasks (optimization methods, Bayesian framework, linear algebra, analytic geometry)
- Strong experience in ML frameworks (preferably PyTorch)
- Experience in digging through the latest scientific papers and identifying applicability for a particular problem
- Strong interpersonal, oral, written, and visual communication skills, with ability to present findings concisely and effectively
- English B2 or higher
The following skills would be a plus:
- Experience in 3D ML problems is a huge plus (e.g., pose estimation, scene reconstruction, surface/volume segmentation, etc.)
- Hands-on experience with in-depth exploration and customization of LLMs, including fine-tuning, embedding generation and optimization, retrieval-augmented generation (RAG) system development, and model evaluation, is a strong plus
- Scientific publications
- Experience in С++ is a plus
- Experience in ML Ops stack (e.g. Kubernetes, AWS cloud)
- Knowledge of SQL