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 in London 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.
