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 to advance the state-of-the-art in weather prediction and large-scale physical simulations more generally.
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 optimisation 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, TensorFlow, Jax).
- 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 physics foundation models!