Senior ML Engineer – Scalable AI on Specialized Hardware

Applied Intuition Inc.; Greater London; On-site; GBP 70,000 - 90,000; Full time

Job summary

A pioneering AI technology firm in London is seeking a Senior Machine Learning Engineer to optimize AI models for their advanced hardware. You will work closely with Software and Research teams to enhance performance and scalability. Ideal candidates should have experience with deep learning frameworks, strong programming skills in Python or C++, and a background in machine learning. The role offers flexible working, competitive salary, and many benefits including private medical insurance and a generous parental leave policy.

Benefits

Flexible working

Private medical insurance

Generous annual leave policy

Employee assistance programme

Qualifications

• BSc/MSc/PhD or equivalent experience in relevant fields.
• Proficiency in deep learning frameworks like PyTorch/JAX.
• Expertise in deep learning from model training to optimisation.

Responsibilities

• Implement and optimise ML models for performance and accuracy.
• Test and evaluate new software releases, provide feedback, and conduct code reviews.
• Benchmark models to identify and improve performance bottlenecks.

Skills

Deep learning frameworks (PyTorch/JAX)

Python or C++ software development

Distributed training or inference across multiple accelerators

Strong communication skills

MLOps for Kubernetes-based clusters

Experience writing C++/Triton/CUDA kernels

Education

Bachelor's/ Master's/ PhD in Machine Learning, Computer Science, Maths, Data Science, or related field

Tools

HPC systems and networking (Infiniband, NVLink, RoCE technologies)

Cloud computing platforms

Job description

A pioneering AI technology firm in London is seeking a Senior Machine Learning Engineer to optimize AI models for their advanced hardware. You will work closely with Software and Research teams to enhance performance and scalability. Ideal candidates should have experience with deep learning frameworks, strong programming skills in Python or C++, and a background in machine learning. The role offers flexible working, competitive salary, and many benefits including private medical insurance and a generous parental leave policy.