Software Engineer - ML Frameworks

Key Skills

AWS, Azure, Deep Learning, Effective communication skills, Google Cloud Platform (GCP), JAX framework, Machine learning techniques, Python Programming, PyTorch, TensorFlow

Job Description

Founded in 2014, InstaDeep is today an EMEA leader in decision-making AI products for the Enterprise, with headquarters in London, and offices in Paris, Berlin, Tunis, Lagos, Dubai, Cape Town and the USA. InstaDeep has been named among the Top 100 global AI startups for three consecutive years by CB Insights, as well as one of the 100 most promising B2B companies in Europe, by Sifted. We have created an environment that ensures a challenge by working closely with a broad spectrum of high-quality clients, pushing you to thrive in an environment that’s rarely tedious and always looking to push you to come up with outside-the-box solutions using cutting-edge technologies. In our mission to stay ahead of the curve, we are proud to partner with firms such as BioNTech SE, Google DeepMind, NVIDIA and Intel, and world-class universities such as MIT, Oxford, UCL, and Imperial College London. We are also an NVIDIA Elite Service Delivery Partner and Google Cloud Partner.

Are you a Software Engineer interested in helping us develop our next generation of open-source machine learning (ML) libraries? The ideal candidate will be passionate about building cutting-edge frameworks and having an impact across InstaDeep’s ML activities. The candidate will help design, develop and maintain our growing ecosystem of ML libraries. Additionally, they will help influence and craft the direction of InstaDeep's engineering culture and mentor both researchers and software engineers. Part of this role will also involve close collaboration with research scientists and engineers for adapting our latest research innovations into the InstaDeep ecosystem.

If so then keep reading….

As a Software Engineer - ML Frameworks you’ll be a part of the Research department, The goal of the Frameworks team is to build high-quality tools, libraries and frameworks to empower the latest research and innovations in deep learning and decision-making. As a Software Engineer in this team, you can help us to develop a cutting-edge software ecosystem to deliver new products, services and research breakthroughs that were previously out of reach.

The Software Engineer - ML Frameworks must accomplish the following: 

Designing and developing our internal and open-source libraries in deep learning and reinforcement learning.

Designing and implementing systems for high-performance computing and for training large models at scale on hardware accelerators, e.g. GPUs, TPUs.

Write high-quality, maintainable and modular software libraries and work side-by-side with our researchers and engineers.

Directly influence InstaDeep’s engineering culture and help scale our research to best leverage our hardware capacities.

Mentor junior engineers and lead our effort to reach the Frameworks team’s objectives.

Who you are:

A passionate engineer with a solid background in algorithms, data structures, and distributed systems.

You are proficient in one or more modern programming languages (Python, C++, Java)

You have experience with at least one popular deep learning framework (Jax, PyTorch, Tensorflow).

Proficiency with Git workflow or similar development with best coding practices.

Software engineering experience demonstrated via contributions to open source, work experience or coding competitions.

You have strong analytical skills and can formulate a logical approach to problem-solving and planning your work.

Strong communication skills and experience collaborating with researchers and developers.

Excellent English skills.

In addition, the following would be an advantage:

Experience with high-performance computing methodologies.

Understanding of modern deep learning methods such as Transformer models, Reinforcement Learning, Meta-Learning or Population Based Training.

Understanding and experience with MLOps practices and public cloud environments (GCP, AWS, Azure).