Senior ML Frameworks Engineer — Research & Deployment

Our research team publishes advanced research on reinforcement learning in top AI conferences such as NeurIPS and collaborates with world-leading researchers and companies.

From Q2 2022, Insta Deep is expanding in the United States. With this aim, Insta Deep is looking for a ML Software/Dev Ops Engineer to support the development of its US activities. J

In this role at Insta Deep, you will report to the Research Lead. You will design and create the algorithms capable of learning and making predictions that define machine learning.

KEY RESPONSIBILITIES

• Design, implement and deliver performant and scalable machine learning libraries.
• Perform computationally intensive tasks with large and complex data using distributed computing systems (CPUs, GPUs, TPUs, Cloud, etc.).
• Write and maintain high-quality, maintainable and modular code with concise documentation and continuous integration together with research scientists and engineers.
• Contribute to the design and implementation of models, tools and libraries to accelerate our research effort.
• Bridge the gap between the research and product teams by integrating new fundamental research into applied projects.

SKILLS AND EXPERIENCE

• S./Ph.D. degree in Computer Science, Machine Learning or related field.
• Experience using Deep Learning frameworks like PyTorch, Tensorflow and/or Jax.
• Experience in developing and debugging in Python3 and in at least one other compiled language (e.g. C++). Strong Software engineering experience (Object-Oriented Programming, Unit Testing, Profiling, CI, Docker) via previous work or contributions to open-source projects.
• Experience with high-performance computing methods and frameworks such as MPI, and Ray (distribution on clusters of CPUs and accelerators) Experience with Dev Ops and cloud technologies GCP, Docker, and Kubernetes.

NICE-TO-HAVES

• 2-5 years' experience in Reinforcement Learning, Computer Vision, or Natural Language Processing experience.
• Structural biology or molecular dynamics experience.
• Published scientific papers in domains in relation to Artificial Intelligence and/or Bio Informatics