ML Infrastructure Engineer — Multi-Node GPU Training

ML Infrastructure Engineer — Multi-Node GPU Training

Full-Time On-site
A

A London-based robotics startup is seeking a Research Engineer to own the scaling of training infrastructure, improving speed and compute efficiency. You will work closely with researchers and tackle distributed systems challenges in a hands-on, early-stage environment.

Strong Python/PyTorch skills and experience with multi-node training are essential, with exposure to NCCL, FSDP or DeepSpeed and AWS cloud GPUs preferred. The role emphasizes ownership and fast learning.

#J-18808-Ljbffr

ML Infrastructure Engineer — Multi-Node GPU Training employer: Axiōma Search

Axiōma Search is an exceptional employer for those passionate about machine learning and time-series forecasting. With a collaborative work culture that prioritises innovation and employee growth, team members are encouraged to explore new ideas and technologies while benefiting from a global network of experts. Located in Europe, the company offers unique opportunities for professional development and the chance to make a significant impact in the field of AI.

A

Contact Details:

Axiōma Search Recruitment Team