Research Engineer in AI

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 Role:We seek a Research Engineer to join our research team and play a key role in advancing our mission to build cutting-edge AI models for decision-making with application to large-scale industrial optimisation. You will be involved in all aspects of the research process, including:Developing and implementing high-quality Python code in a production-level environment.Distributing and accelerating code on large-scale accelerator clusters.Identifying and preprocessing relevant data sources, and building efficient data loading pipelines.Designing and executing training and validation pipelines, and interpreting results.Contributing to the development of research plans and disseminating our work through research papers and participation in conferences.We are a results-oriented, hands-on team that thrives on collaboration. You will have the opportunity to work closely with fellow researchers, applied scientists, and product teams.

Responsibilities:

• 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.
• Algorithmic Design: Identify limitations in the application of existing state-of-the-art algorithms for our desired use cases, and propose concrete algorithmic contributions to mitigate those limitations and improve upon existing state-of-the-art.
• Scaling Expertise: Design and implement strategies to efficiently scale machine learning models across different accelerator 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.

Required Skills:

• A Masters or equivalent in Computer Science, Engineering, Mathematics, or related field.
• Fundamentals of modern Deep Learning and experience with Reinforcement Learning.
• 2+ years of work experience.
• Expertise in Python.
• Experience working with Linux systems.
• Experience with at least one modern machine learning framework (JAX, Tensorflow, PyTorch, etc.)
• Experience with Docker.

Highly Desirable:

• Experience working with JAX and packages within the JAX ecosystem.
• Track record of successfully building and scaling ML models.
• Experience with software profiling, identifying bottlenecks, and delivering efficient solutions.

Our commitment to our peopleWe 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.