Senior Machine Learning Engineer (ML Platform)

Join DeepHealth as a Senior Machine Learning Engineer, where you will be responsible for building, running, and evolving the ML monorepo used by the X-ray, CT, and MRI machine learning group. You will own the architecture, tooling, and engineering standards of the codebase, and lead the team's agentic coding strategy. This hands-on role requires experience in taking models to production, solid PyTorch knowledge, and the ability to work autonomously in a distributed team.

Follows all local, regional and country laws concerning employment

Follows data privacy, compliance, safety and confidentiality standards at all times

Follows all DeepHealth policies and procedures

Reports to work regularly as scheduled; consistently punctual with respect to working hours, meal and rest breaks, and maintains satisfactory personal attendance in accordance with DeepHealth guidelines

Responsibilities

• Own the architecture and roadmap of the ML Monorepo, including module boundaries, shared abstractions, and experiment tracking.
• Build and run the engineering tooling around the codebase, including continuous integration, testing, dependency management, and GPU job scheduling.
• Drive the team's agentic coding strategy, defining how AI coding agents are used in the monorepo, and setting up the tooling, guardrails, and conventions.
• Adapts and is willing to learn new tasks, methods, and systems
• Demonstrates initiative, personal awareness, professionalism and integrity, and exercises confidentiality in all areas of performance
• Completes job responsibilities in a quality and timely manner
• Practices universal safety precautions
• Communicates, cooperates, and consistently functions professionally and harmoniously with all levels of supervision, co-workers, visitors, and vendors

Requirements

• Experience taking models to production: model packaging, inference serving, and performance and reliability constraints (required)
• Ability to work autonomously in a distributed, multi-site team, with clear verbal and written communication in English (required)
• 5+ years of software engineering experience, including a significant portion building infrastructure or tooling used by machine learning teams, such as an ML platform or a training framework (required)
• Strong Python skills and software engineering fundamentals: API design, testing, refactoring, and attention to developer experience (required)
• Master's degree in computer science, engineering, mathematics, or a related field, or equivalent practical experience (required)
• Ability to work closely with researchers, translate recurring pain points into shared tooling, and push back constructively when needed (required)
• Promotes good public relations on the phone and in person

Preferred

• Solid PyTorch knowledge and practical experience with model training and profiling (required)
• Contributions to open-source ML tooling (preferred)
• Experience running ML workloads on a cloud platform, ideally GCP (Vertex AI, Artifact Registry, IAM, storage), and with containerization (required)
• HPC and Slurm experience (preferred)
• Medical imaging experience (DICOM, 3D volumes, large images) (preferred)

Pay

EUR 50,000 - 80,000