Machine Learning Research Engineer / Scientist

Job Summary

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.

Responsibilities

• Arcitect 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.
• 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.

Requirements

• 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.

Qualifications

US citizen/visa only.

Benefits

Competitive salary range: $120K - $250K, with a typical offer around $180K.

Pay

Base Salary: $120K - $250K

Schedule

Full-time