Research Engineer, Data Infrastructure

Mistral AI•Palo Alto, CA

About The Position

This role focuses on building and operating the next generation of data infrastructure at Mistral AI. You will be a core contributor to our evolution, helping us design and scale massive compute fleets and storage systems designed for high performance and scalability. You will help us move toward a future of decoupled control and data planes, scaling big data compute and storage platforms while ensuring secure and governed data access for MLOps and research. You will take full lifecycle ownership: from architecting the migration away from legacy orchestrators to implementing production-grade pipelines and participating in on-call rotations for critical training jobs.

Requirements

Have 4+ years of experience in Data Infrastructure, MLOps, or Infrastructure Engineering.
Have experience or a strong interest in supporting foundational compute and storage platforms.
Are proficient in Python and enjoy solving the "brittle data lake" problem with modern, columnar storage standards.
Are well-versed in Kubernetes-native tooling and excited to debug large-scale distributed systems across multi-cluster environments.
Take pride in building and operating scalable, reliable, and secure systems from the ground up.
Are comfortable with ambiguity and the challenges of building high-scale infrastructure in a rapid-growth AI environment.

Responsibilities

Build & Scale: Help us reach our goal of operating massive distributed compute and storage systems
Global Orchestration: Architect and maintain multi-cluster orchestration layers to optimize workload placement across diverse hardware and regions.
Design Future-Proof Storage: Architect our transition to modern storage formats to handle fine-tuning datasets at a scale that anticipates exabyte growth.
Platform Engineering: Contribute to the development of our internal training platform, ensuring seamless model training and fine-tuning capabilities across Kubernetes and SLURM based environments.
Metadata & Lineage : Implement and manage systems to provide clear visibility and lineage as our data and model pipelines grow in complexity.
Operational Excellence : Use modern deployment workflows to manage cloud-native deployments, ensuring our data platform can scale.

Benefits

Competitive salary and equity.
Medical/Dental/Vision covered for you and your family.
401K (6% matching)
18 days PTO
Reimburse office parking charges, or $120/month for public transport
$120/month reimbursement for gym membership
$400 monthly allowance for meals
Visa sponsorship
BetterUp coaching on a voluntary basis
Healthcare Coverage 100% employer-sponsored premium plans for medical, dental, and vision care for you and your dependents.
Parental Leave 20 weeks of paid leave for all primary parents.
Childcare Support Reserved daycare seats or financial assistance for working parents.
Retirement Plans Competitive employer-matched contributions in applicable markets (e.g., 6% 401k match in the US, 5% pension contribution in the UK) to secure your financial future.
Relocation Support Financial assistance for moving expenses, visa sponsorship, and settling-in services for you and your family.
Daily Life & Wellness Meal Allowances Monthly stipends, vouchers, or catered in-office meals, depending on your office location.
Transportation Support Monthly allowances for public transport, parking, or sustainable mobility
Fitness & Wellness Subsidies or memberships for gyms, wellness programs, or fitness activities