Machine Learning Research Engineer / Scientist

Qualifications

• Proven track record in developing and deploying deep learning models
• Advanced proficiency in Python and modern ML frameworks (PyTorch, Jax or similar)
• Demonstrated experience with distributed training systems and large-scale data pipelines
• Strong software engineering practices and system design principles
• Excellent problem-solving and analytical skills
• Outstanding communication and collaboration abilities
• MSc or PhD in Artificial Intelligence, Computer Science, or related technical field
• Published research in prestigious AI conferences/journals (NeurIPS, ICML, etc.)
• Hands-on experience with one or several of the following: transformers, diffusion models, self-supervised learning, foundation model training/fine-tuning

Responsibilities

• Architect and implement innovative ML models for complex spatiotemporal data analysis
• Lead end-to-end development of large-scale AI systems, from research to production
• Drive the optimization of training and inference pipelines for maximum performance
• Conduct validation experiments and performance analysis
• Spearhead long-term research initiatives with significant real-world impact
• Collaborate with world-class researchers and engineers

Full Description

Role Overview We are seeking Machine Learning Research Engineers / Scientists to join our team working on groundbreaking physics foundation models.

The successful candidate will develop, train and deploy to production large-scale AI foundation models for weather, energy, and beyond.

What You'll Do • Architect and implement innovative ML models for complex spatiotemporal data analysis.

• Lead end-to-end development of large-scale AI systems, from research to production.

• Drive the optimization of training and inference pipelines for maximum performance.

• Conduct validation experiments and performance analysis.

• Spearhead long-term research initiatives with significant real-world impact.

• Collaborate with world-class researchers and engineers. We expect you to have • Proven track record in developing and deploying deep learning models.

• Advanced proficiency in Python and modern ML frameworks (PyTorch, Jax or similar).

• Demonstrated experience with distributed training systems and large-scale data pipelines.

• Strong software engineering practices and system design principles.

• Excellent problem-solving and analytical skills.

• Outstanding communication and collaboration abilities. Nice to have • MSc or PhD in Artificial Intelligence, Computer Science, or related technical field.

• Published research in prestigious AI conferences/journals (NeurIPS, ICML, etc.).

• Hands-on experience with one or several of the following: transformers, diffusion models, self-supervised learning, foundation model training/fine-tuning. Join us in pushing the boundaries of foundation models for the physical world!