Requirements

• Completed MSc in computer science, machine learning, AI or a related area
• Proficiency in collaborative software engineering in Python
• In-depth understanding of machine learning frameworks such as PyTorch and/or Jax
• Experience in designing, developing and deploying ML systems
• Experience with agentic software development
• Experience building and optimizing distributed systems and large-data applications, including those using tensor accelerators or GPUs
• Ability to work in an interdisciplinary collaborative environment, effectively communicating technical concepts to non-experts

Nice to haves

• PhD degree in computer science, machine learning, AI or a related field, or comparable industry experience with machine learning and large datasets
• Experience working with major cloud platforms and/or HPC
• Experience developing high-performance scientific software
• Experience designing or developing agentic frameworks, orchestration systems, or infrastructure for building and running AI agents

What you'll be doing

• Develop and maintain efficient, reliable, and scalable infrastructure for large-scale machine learning training, evaluation, and benchmarking
• Architect, design, and implement large-scale data pipelines for heterogeneous scientific data, supporting model training, experimentation, evaluation, and benchmarking
• Build and optimize distributed data processing, model training, evaluation, and benchmarking pipelines
• Develop software and agentic workflows to automate and improve model development, training, evaluation, and benchmarking
• Continuously monitor and improve the stability of data, training, evaluation, and benchmarking workflows
• Work cross-functionally with machine learning researchers, engineers, and natural science researchers
• Maintain high standards in code quality and software design
• Document and share best practices across the organization

Perks and benefits

• Competitive base pay ranges depending on location (e.g. £74,700.00 - £122,600.00 in the UK; €90,100.00 - €152,500.00 in Germany; €78,800.00 - €133,300.00 in the Netherlands)
• Potential eligibility for additional benefits and other compensation
• Equal opportunity employer with support for religious and disability accommodations during the application process