Machine Learning Engineer, Research

What you will do

• Work intimately with our simulation engineers and research scientists to develop an understanding of the physics and engineering challenges we are solving
• Design, build and optimise machine learning models with a focus on scalability and efficiency in our application domain
• Transform prototype implementations to robust production-grade implementation of models
• Explore distributed training architectures and federated learning capacity
• Create analytics environments and resources in the cloud or on-premise, spanning data engineering and science
• Identify the best libraries, frameworks and tools for our modelling efforts to set us up for success
• Work at the intersection of data science and software engineering to translate the results of our R&D into re-usable libraries, tooling and products
• Continuously apply and improve engineering best practices and standards and coach your colleagues in their adoption

What you bring to the table

• Enthusiasm about using machine learning for science and engineering, and especially in scaling such solutions to real-world settings
• Degree (Master's/Doctorate) in computer science, software engineering or equivalent
• Experience scaling ML models, both in compute and data storage
• Federated learning experience is a bonus
• 1+ year of experience in a data-driven role, with exposure to:
• Software engineering concepts and best practices (e.g., versioning, testing, CI/CD, API design, MLOps)
• Building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., PyTorch, MLFlow, JAX)
• Distributed computing frameworks (e.g., Spark, Dask)
• Cloud platforms (e.g., AWS, Azure, GCP) and HP computing
• Containerization and orchestration (Docker, Kubernetes)
• Ability to scope and effectively deliver projects
• Strong problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly
• Excellent collaboration and communication skills - with teams and especially researchers

What we offer

• Be part of something larger: Make an impact and meaningfully shape an early-stage company. Work on some of the most exciting and important topics there are. Do something you can be proud of
• Work with a fun group of colleagues that support you, challenge you and help you grow. We come from many different backgrounds, but what we have in common is the desire to operate at the very top of our fields and solve truly challenging problems in science and engineering. If you are similarly capable, caring and driven, you'll find yourself at home here
• Experience a truly flat hierarchy. Voicing your ideas is not only welcome but encouraged, especially when they challenge the status quo
• Work sustainably, striking the right balance between work and personal life.
• Receive a competitive compensation and equity package, in addition to plenty of perks such as generous vacation and parental leave, complimentary office food, as well as fun outings and events
• Work in a flexible setting, at our lovely London Shoreditch office, and a good proportion from home if so desired. Get the opportunity to occasionally visit our customers' engineering sites and experience first-hand how our work is transforming their ways of working
• Use first-class equipment for working in-office or remotely, including HPC