ML Research Engineer: Open-Source AI Platform & Equity

Mistral AI; Palo Alto (CA); On-site; USD 120,000 - 150,000; Full time

Benefits offered by this job

Competitive salary and equity

Healthcare: Medical/Dental/Vision covered

Pension: 401K (6% matching)

PTO: 18 days

Transportation reimbursement

Gym membership reimbursement

Meal stipend of $400 monthly

Visa sponsorship

Coaching services

Job summary

A pioneering AI company in California is seeking a Research Engineer for ML to enhance large-scale learning systems and collaborate with Research Scientists. The ideal candidate will have a Master's or PhD in Computer Science, over four years of experience in ML codebases, and skills in Python, PyTorch, or TensorFlow. This position offers competitive salary and equity, healthcare coverage, a 401K plan, generous PTO, and other benefits. Join the team to innovate and drive your career in AI forward.

Qualifications

• 4+ years working on large-scale ML codebases.
• Experience with distributed training.
• Strong software-design instincts.

Responsibilities

• Accelerate researchers by building robust tools.
• Integrate cutting-edge research with production.
• Conduct experiments on the latest deep-learning techniques.

Skills

ML pipelines

Python

PyTorch

TensorFlow

JAX

Collaborative teamwork

Education

Master’s or PhD in Computer Science

Tools

DeepSpeed

FSDP

K8s

Job description

A pioneering AI company in California is seeking a Research Engineer for ML to enhance large-scale learning systems and collaborate with Research Scientists. The ideal candidate will have a Master's or PhD in Computer Science, over four years of experience in ML codebases, and skills in Python, PyTorch, or TensorFlow. This position offers competitive salary and equity, healthcare coverage, a 401K plan, generous PTO, and other benefits. Join the team to innovate and drive your career in AI forward.