Jobs

We’re hiring a Machine Learning Engineer to work on lightweight text classification and embedding models operating on edge devices under tight memory and latency constraints. You will own the entire model development lifecycle, from data labeling to model training and deployment.

What you’ll do

• Design evaluation metrics and benchmarks.
• Build data processing and labeling pipelines.
• Experiment with model architectures and training methods.
• Contribute to our training and model optimization framework

What we’re looking for

• Strong background in Deep Learning, with deep knowledge of modern text foundation models
• Solid grasp of statistics and probability theory
• Understanding of ML performance optimization landscape
• Experience building complex training pipelines from scratch
• Very strong software engineering skills.

Bonus:

• Experience with JAX
• Experience with GPU programming
• Knowledge of Rust

You are probably a good fit if you have a thorough understanding of such concepts as:

• muP
• Muon
• FSDP
• FlashAttention
• SigLIP

Why join?
You’ll work on applied research that directly impacts how AI systems operate in real-world environments. Not just benchmarks.
We value technical depth, fast iteration, and clarity of thought.
Competitive compensation + meaningful equity.

About us
Mirai builds advanced on-device AI systems. We operate at the intersection of model design and hardware-aware optimization.
Our Models & Research team focuses on making small, efficient models reliable under strict real-world constraints.