Staff AI Research Engineer - ML Engineering

Berlin; London; Munich. Posted 9 months ago

The role

You will architect and implement tools and platforms for autonomous decision-making in defence. You will extend deep learning frameworks, scale infrastructure for distributed training, and design data strategies for large scale datasets.

What you will do

Extend highly integrated deep learning frameworks built on top of PyTorch.
Scale current infrastructure and tooling stack to support faster and larger distributed training.
Design data strategy to support large scale datasets and efficient storage.

Required qualifications

MSc or PhD in Computer Science or STEM field, with a focus on Machine Learning and Deep Learning.
Strong software engineering skills in Python.
Fluency with modern DL frameworks (PyTorch/JAX/TensorFlow).
Ability to write custom layers, loss functions, and distributed training loops.
Clear communicator.
First-principles mindset.
Experience debugging production ML pipelines.

Nice to have

Hands-on experience training models on large-scale GPU clusters.
Implementing advanced parallelism strategies.
Understanding underlying cross-node communication patterns (NCCL, MPI).
Worked with large-scale datasets made of different modalities.
Understanding tradeoffs in locality, encoding, formats, and streaming strategies.
Proficient with workload orchestrators like Slurm, Kubernetes, or Ray.
Understand GPU architecture at a low level.

Benefits

Competitive salary
VSOP options
Relocation support: up to €2,500 and 4 weeks temporary accommodation
Learning: €500/£450 yearly allowance
Gym membership
Mental health support (Nilo.health)
Regular company events
Monthly social allowances
22 weeks fully paid parental leave for primary caregivers
6 weeks parental leave for secondary caregivers
5 days of paid family emergency leave
100% remote work option during pregnancy and phased return to work

About the company

Helsing is a defence AI company. Their mission is to protect democracies through technological leadership.