Senior Machine Learning Engineer, Scaling and Performance

Join to apply for the Senior Machine Learning Engineer, Scaling and Performance role at InstaDeep

Senior Machine Learning Engineer, Scaling and Performance

Join to apply for the Senior Machine Learning Engineer, Scaling and Performance role at InstaDeep

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:

Our team plays a pivotal role in enhancing the capabilities and efficiency of our advanced AI systems. We design solutions that enable our machine learning models to scale seamlessly and perform optimally in real-world applications and large scale research. Collaborating across InstaDeep, we directly impact projects in diverse fields including Life Sciences, Logistics, Chip Design, and Quantum ML.

The Role:

We seek a highly skilled Machine Learning Engineer with a passion for tackling the challenges of large-scale ML development. You'll play a vital role in making our ambitious AI solutions a practical reality. If you thrive on system-level analysis, find joy in squeezing every ounce of performance from hardware, and love diving deep into algorithm optimisation, this is the position for you.

Responsibilities

• Scaling Expertise: Design and implement strategies to efficiently scale machine learning models across diverse hardware platforms (GPU/TPU).
• Performance Optimisation: Analyse and profile ML systems under heavy load, pinpointing bottlenecks, and implementing targeted optimisations.
• Distributed Systems Architecture: Create robust distributed training and inference solutions for maximum computational efficiency.
• Algorithmic Optimisation: Research and understand the latest deep learning literature to implement and optimise state-of-the-art algorithms and architectures, ensuring compute efficiency and performance.
• Low-Level Mastery: Write high-quality Python, C/C++, XLA, Pallas, Triton, and/or CUDA code to achieve performance breakthroughs.
  
  

• Understanding of Linux systems, performance analysis tools, and hardware optimisation techniques
• Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning, etc.)
• Expertise with Python and/or C/C++
• Development with machine learning frameworks (JAX, Tensorflow, PyTorch etc.)
• Passion for profiling, identifying bottlenecks, and delivering efficient solutions.
  
  

• 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.
• Fundamentals of modern Deep Learning
• Actively following ML trends and a desire to push boundaries.
  
  

• Profile algorithm traces, identifying opportunities for custom XLA operations and CUDA kernel development.
• Implement and apply SOTA architectures (MAMBA, Griffin, Hyena) to research and applied projects.
• Adapt algorithms for large-scale distributed architectures across HPC clusters.
• Employ memory-efficient techniques within models for increased parameter counts and longer context lengths.
  
  

• Real-World Impact: Directly contribute to the performance and reach of our AI solutions.
• Cutting-Edge Challenges: Tackle complex problems at the forefront of machine learning and large-scale system design.
• Growth-Oriented Environment: Expand your expertise in a team of talented engineers dedicated to advancing ML scalability.
  
  

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Engineering and Information Technology

• Industries

  Software Development

Referrals increase your chances of interviewing at InstaDeep by 2x

Sign in to set job alerts for “Machine Learning Engineer” roles.

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 1 month ago

London, England, United Kingdom 19 hours ago

London, England, United Kingdom 2 days ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 1 month ago

London, England, United Kingdom 5 months ago

London, England, United Kingdom 2 days ago

Machine Learning Scientists and Engineers: AI for Quantum

London, England, United Kingdom 2 months ago

London, England, United Kingdom 2 weeks ago

Greater London, England, United Kingdom 4 weeks ago

London, England, United Kingdom 4 weeks ago

London, England, United Kingdom 1 month ago

London, England, United Kingdom 5 days ago

Machine Learning Engineer - Search and Recommendation

London, England, United Kingdom 1 month ago

London, England, United Kingdom 4 days ago

Graduate Software Engineer 2025 - RegTech

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 2 months ago

London, England, United Kingdom £50,000.00-£60,000.00 1 month ago

London, England, United Kingdom 2 days ago

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 20 hours ago

London, England, United Kingdom 1 day ago

London, England, United Kingdom 2 months ago

London, England, United Kingdom 3 weeks ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 5 months ago

Greater London, England, United Kingdom 3 weeks ago

Research Engineer, ML, AI & Computer Vision

London, England, United Kingdom 2 weeks ago

City Of London, England, United Kingdom 4 days ago

London, England, United Kingdom 3 weeks ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.