Applied Scientist / Research Engineer

MISTRAL AI PTE. LTD.; Singapore; On-site; SGD 80,000 - 120,000; Full time

Job summary

A leading AI research firm in Singapore is hiring Applied Scientists and Research Engineers to develop innovative AI solutions. Candidates should be fluent in English and expert in PyTorch or JAX, with a strong background in mathematics or machine learning. Responsibilities include developing state-of-the-art models and managing research projects. The role offers a competitive salary and several perks including health insurance and gym allowances.

Benefits

Competitive cash salary and equity

Health Insurance

Gym membership allowance

Monthly allowance for meals

Transportation allowance

Qualifications

• Expertise in handling state of the art models.
• Track record in personal, professional, or academic projects.

Responsibilities

• Run pre-training and deploy models on GPU clusters.
• Generate and curate data for model training.
• Develop tools for data generation and model evaluation.
• Manage research projects and client communications.

Skills

Fluent English

Expert with PyTorch or JAX

High-performance Python code

Clear communication of technical concepts

Collaborative and low-ego

Education

PhD or Master's in Mathematics, Physics, or Machine Learning

Job description

About The Job

Mistral AI is seeking Applied Scientists and Research Engineers to drive innovative research and collaborate with clients on complex research projects.

You will develop SOTA models across different modalities such as text, image, and speech. By developing novel methods and research ideas you will apply these models across a diverse set of use cases and domains. Working cross-functionally with both external and internal science, engineering, and product teams you will deliver high-impact AI solutions that turn the needle.

What you will do

• Run pre-training, post-training and deploy state of the art models on clusters with thousands of GPUs. You don’t panic when you see OOM errors or when NCCL feels like not wanting to talk.
• Generate and curate data for pre-training and post-training, working on evaluations and making sure the model’s performance beats expectations.
• Develop the necessary tools and frameworks to facilitate data generation, model training, evaluation and deployment.
• Collaborate with cross-functional teams to tackle complex use cases using agents and RAG pipelines.
• Manage research projects and communications with client research teams.

About you

• You are fluent in English, and have excellent communication skills. You are at ease explaining complex technical concepts to both technical and non-technical audiences.
• You’re an expert with PyTorch or JAX.
• You’re not afraid of contributing to a big codebase and can find yourself around independently with little guidance.
• You write clean, readable, high-performance, fault-tolerant Python code.
• You don’t need roadmaps: you just do. You don’t need a manager: you just ship.
• Low-ego, collaborative and eager to learn.
• You have a track record of success through personal projects, professional projects or in academia.

It would be great if you

• Hold a PhD / master in a relevant field (e.g., Mathematics, Physics, Machine Learning), but if you’re an exceptional candidate from a different background, you should apply.
• Can bring a variety of research experience (agents, multi-modality, robotics, diffusion, time-series).
• Have contributed to a large codebase used by many (open source or in the industry).
• Have a track record of publications in top academic journals or conferences.
• Love improving existing code by fixing typing issues, adding tests and improving CI pipelines.

Benefits

• 💰 Competitive cash salary and equity
• 🚑 Health Insurance
• 🥎 Sport : $90 for gym membership allowance
• 🥕 Food : $200 monthly allowance for meals (solution might evolve as we grow bigger)
• 🚴 Transportation : $120/month for public transport or Parking charges reimbursed

We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.