AI Machine Learning Engineer

At Mirai, we work with incredible companies looking for exceptional engineers, and this is one of the most technically ambitious opportunities we’ve partnered on.

We’re working with a world-leading AI research organisation building the infrastructure behind next-generation AI systems used to accelerate scientific discovery. This isn’t a traditional product company or biotech startup—it’s an opportunity to build the machine learning platforms and distributed systems powering some of the largest AI workloads in science.

If you enjoy solving complex infrastructure challenges, optimising large-scale training systems, and building platforms that enable researchers to push the boundaries of AI, we’d love to speak with you.

The Opportunity

• Join an elite engineering team building infrastructure for frontier AI research
• Build large-scale machine learning systems that support cutting-edge foundation models
• Design and optimise GPU-native training and data loading pipelines
• Work on distributed systems operating across petabyte-scale datasets
• Build reliable infrastructure that enables researchers to move faster
• Help shape engineering standards and core platform architecture
• Collaborate closely with world-class AI researchers and infrastructure engineers
• Work on production systems that directly enable scientific breakthroughs

What You Will Work On

• Building scalable machine learning infrastructure for large-scale model training
• Optimising GPU utilisation, distributed training, and high-performance data pipelines
• Developing production MLOps systems covering training, deployment, monitoring and reliability
• Designing efficient storage, I/O and data loading solutions for large scientific datasets
• Building infrastructure supporting model lifecycle management and experiment tracking
• Improving engineering abstractions to accelerate AI research
• Developing CI/CD, testing and platform tooling for research teams
• Collaborating across engineering and research to productionise frontier AI models

What We Are Looking For

• 5+ years of experience building production machine learning infrastructure
• Strong Python and PyTorch experience
• Experience with distributed training and GPU optimisation
• Familiarity with large-scale data formats such as Zarr, HDF5, TensorStore or similar
• Experience with distributed computing frameworks such as Spark, Dask or Ray
• Experience with Docker and Kubernetes
• Strong software engineering fundamentals and a passion for building infrastructure that others depend on
• Comfortable operating in highly technical, research-driven environments

Useful Experience

• Building large-scale AI or deep learning infrastructure
• Performance optimisation across GPU clusters
• MLOps and production ML platform engineering
• AI agent frameworks
• Scientific computing or high-performance computing environments
• Experience supporting large-scale distributed training workloads

Role Details

Location: New York City (Hybrid)

Compensation: Up to $235,000 base salary + comprehensive benefits

Type: Permanent, Full-Time

Diversity & Inclusion

Mirai believes diverse teams build better companies and better technology. We welcome applications from individuals of all backgrounds and are committed to building inclusive environments where people can do their best work.