InstaDeep

Lead Machine Learning Engineer

United Kingdom

Hybrid

Full Time

#Research

#Python

#C C++

#Linux Systems

#Performance Analysis

#Machine Learning

#Jax

#TensorFlow

#PyTorch

#CUDA

InstaDeep is looking for a Lead Machine Learning Engineer

Full Time Lead Hybrid United Kingdom Research Python C C++Linux Systems Performance Analysis Machine Learning Jax TensorFlow PyTorch CUDA English Hybrid work

InstaDeep, founded in 2014, is a pioneering AI company at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. We have been listed among notable players in AI, fast-growing companies, and Europe's 1000 fastest-growing companies in 2022 by Statista and the Financial Times. Our recent acquisition by BioNTech has further solidified our commitment to leading the industry.

Join us to be a part of the AI revolution!

The Team

Efficiently training machine learning algorithms at scale requires solving novel system problems. Our team leads the design and implementation of high-performance solutions to seamlessly scale our AI systems, including our latest foundation models in biology and beyond. We optimise throughput, scalability, and robustness in some of the largest distributed ML systems, making ambitious research ideas a practical reality.

The Role

We're looking for a Lead Machine Learning Engineer to take charge of tackling performance bottlenecks and lead the development of solutions that scale machine learning to the next level. In this role, you’ll collaborate with a team of software and performance engineers to build systems that enable the next generation of our research. Strong candidates will have demonstrated expertise in managing and executing complex ML system solutions, coupled with a drive to optimise performance and scalability in state-of-the-art workloads.

Responsibilities

Technical Leadership: Define the long-term technical roadmap and drive the development of scalable, high-performance ML systems.

Algorithm Optimisation: Optimise state-of-the-art algorithms and architectures from the latest deep learning research for compute efficiency and performance.

System Scaling: Design strategies for scaling machine learning models across diverse hardware platforms (GPU/TPU) and optimising system performance under heavy load.

Low-Level Optimisation: Write efficient Python, C/C++, XLA, Pallas, Triton, or CUDA code to achieve performance breakthroughs.

ML Systems Design: Architect robust distributed systems for training, deployment, and monitoring, ensuring computational efficiency and scalability.

Data Pipeline Automation: Develop automated pipelines for data processing, model training, validation, and deployment, enabling efficient handling of large datasets.

Team Collaboration: Partner with research, applied, and product teams to build a cohesive software stack supporting key projects.

Mentorship: Guide and mentor the ML engineering team, fostering best practices in coding, testing, and documentation.

Required Skills

Expertise with Python and/or C/C++

Understanding of Linux systems, performance analysis tools, and hardware optimisation techniques.

Development with machine learning frameworks (JAX, Tensorflow, and/or PyTorch)

Passion for profiling, identifying bottlenecks, and delivering efficient solutions.

Fundamentals of modern Deep Learning

Desired Skills

Track record of successfully scaling ML models.

Experience writing custom CUDA kernels or XLA operations.

Understanding of GPU/TPU architectures and their implications for efficient ML systems.

Representative projects

Profile algorithm, identifying opportunities for custom XLA/CUDA kernels.

Implement SOTA architectures (MAMBA, Griffin, Hyena) to research and applied projects.

Adapt algorithms for large-scale distributed architectures across HPC clusters.

What we offer

A chance to lead and grow a team of talented engineers in solving some of AI’s most challenging system problems.

Hands-on experience optimising large-scale distributed ML systems that underpin industry-leading research;

A front-row seat to the evolution of AI, with opportunities to shape its direction through technical innovation and leadership.

TLDR: Lead a team of engineers to design and implement innovative engineering solutions for scaling ML systems, enabling InstaDeep’s most ambitious AI research.

Our commitment to our people

We empower individuals to celebrate their uniqueness here at InstaDeep. Our team comes from all walks of life, and we’re proud to continue encouraging and supporting applicants from underrepresented groups across the globe. Our commitment to creating an authentic environment comes from our ability to learn and grow from our diversity, and how better to experience this than by joining our team? We operate on a hybrid work model with guidance to work at the office 3 days per week to encourage close collaboration and innovation. We are continuing to review the situation with the well-being of InstaDeepers at the forefront of our minds.

Right to work: Please note that you will require the legal right to work without visa sponsorship in the location you are applying for. We do not sponsor work visas.

InstaDeep

4 views

0 applied

Company Size

251-500

Markets

Artificial Intelligence

Information Technology

Social Media

Visit InstaDeep

Share this job

Copy Permalink

Open roles at InstaDeep

InstaDeep

HR Business Partner

France

Hybrid

Full Time

#Finance Operations

#Human Resources

#Payroll Processing

#Benefits Management

#HRIS

#Labor Laws

#Employee Relations

Discover similar jobs

Constructive Dialogue Institute

Senior Data Scientist

United States

135k - 145k USD

Remote

Full Time

#Data Science

#Analytics Engineering

#Nonprofit

#SQL

#Python

#Data Pipelines

#AWS

#Dashboards

#Git

#Data Quality

#BI Tools

Bounce

UX/Product Researcher

Remote

Contractor

#UX Research

#Product Research

#Design

#Research

#Qualitative Research

#Ops

#user

#Coaching

#Communication Skills

Flagright.com

Senior Business Development Representative

Remote

Full Time

#Business Development

#Fintech

#Sales

#Cold Calling

#Outbound Prospecting

#Outreach

#CRM Management

#Pipeline Generation

#Research

#B2B Sales

Tebra

Security Architect

179k - 204k USD

Remote

Full Time

#Security

#Cloud Security

#Healthcare

#Cloudflare

#GCP

#Kubernetes

#Terraform

#Python

#DevSecOps

#Vertex AI

#BigQuery

#Helm

#Workato

Union

Sales Engineer

Remote

Full Time

#AI

#Sales

#Machine Learning

#MLOps

#PyTorch

#TensorFlow

#Spark

#Kubernetes

#Docker

#AWS

#Terraform

#MEDDIC

NewPage Solutions Inc

Python Developer

Remote

Contractor

#Technology

#Digital Health

#Continuous Delivery

#Python

#AWS Lambda

#AWS ECS

#Automated Testing

#Agile Methodologies

#Terraform

#Drupal

#PHP

#S3

#DynamoDB

Deepgram

Pre-Sales Solutions Engineer

Remote

Full Time

#AI

#Solutions Engineering

#Python

#JavaScript

#API Integration

#Speech Recognition

#NLP

#Cloud Platforms

#Docker

#Kubernetes

#Sales Methodologies

HealthEvolution

Business Development Representative

Remote

Full Time

#Sales

#Business Development

#Healthcare

#Salesforce

#Pipeline Management

#Sales Forecasting

#Research

#Data Reporting

#Contract Management

#Scheduling

#Lead Management

saas.group

Applied Research Scientist

Remote

Full Time

#AI

#Research

#SQL

#Python

#Data Analysis

#Experiment Design

#Data Pipelines

#Validation

#AI Tools

#Research Methodology

Dataiku

Fullstack Software Engineer

Remote

Full Time

#Engineering

#AI

#Solutions

#Vue.Js

#React

#Angular

#Python

#fastAPI

#Flask

#RESTful API

#Data

Ankorstore

Lead Data Platform

Remote

Full Time

#Data Engineering

#Platform

#Data Platform

#SQL

#Python

#BigQuery

#Airflow

#DBT

#Sigma

#Amplitude

#Terraform

#Product Analytics

Eight Sleep

Senior Backend Engineer

Remote

Full Time

#Software Engineering

#Distributed Systems

#Java

#Kotlin

#Scala

#C#

#Python

#NodeJS

#TypeScript

#Cloud Services

HTTPie

Senior Fullstack Engineer

Remote

Full Time

#Developer Tools

#Engineering

#API Testing

#TypeScript

#React

#Next.js

#TailwindCSS

#Python

#Django

#AWS

#Terraform

#WebSocket

#Electron

Adthena

Anti-Bot Engineer

Remote

Full Time

#Web Scraping

#Search

#Data Engineering

#Python

#Automation

#Playwright

#Selenium

#Management

#HTTP

#Docker

#Kubernetes

Hummingbird

Senior Software Engineer, Infrastructure

Remote

Full Time

#Infrastructure Engineering

#Fintech

#Security

#AWS

#Terraform

#PostgreSQL

#Redis

#Ruby on Rails

#Python

#Docker

#CircleCi

#GraphQL

#TypeScript

Fullscript

Lead Data Scientist

Remote

Full Time

#Data Science

#Health Tech

#Causal Inference

#Python

#Statistical Modeling

#Research

#Data

#Data Analysis

#Machine Learning

Axelera AI

Field Application Engineering Manager

Remote

Full Time

#AI

#Field Engineering

#Customer Success

#PyTorch

#TensorFlow

#Embedded Systems

#Python

#CUDA

#Computer Vision

#Deployment

eduki

Senior AI Engineer

Spain

39.8k - 57.3k USD

Remote

Full Time

#AI Engineering

#Automation

#Data Analytics

#Systems

#Databases

#Engineering

#AI

#Orchestration

#Python

#SQL

Getty Images

Senior Creative Insights Researcher

Remote

Full Time

#Creative

#Research

#Market Trends

#Data

#PowerPoint

#Excel

#Content Management

#Trend Analysis

Monarchmoney

Engineer

Remote

Full Time

#Personal Finance

#Software Engineering

#React

#TypeScript

#React Native

#Python

#Django

#GraphQL

#PostgreSQL

#AWS

#Terraform

#Docker

Your dream job awaits.

Explore exciting opportunities, connect with top employers, and ignite your career.

Explore Jobs

Related Resources

Research Salary Guide

Compensation data for Research roles

Research Job Market

Hiring trends and demand for Research

Research Interview Prep

Practice questions for Research interviews

Python Salary Guide

Compensation data for Python roles

Python Job Market

Hiring trends and demand for Python

Python Interview Prep

Practice questions for Python interviews

Lead Machine Learning Engineer at InstaDeep