Research Engineer, Training & Inference

Highlights

• Join a team that achieved Gold Medal-level performance at the 2025 IMO.
• Own the full RL stack from simulators to distributed training and inference engines.
• Build systems powering Aristotle, a verified mathematical reasoning engine.

What they offer

• Unlimited PTO
• 401(k) matching
• 100% employer-paid health, vision, and dental benefits for employees and 50% for dependents
• Health Savings Account (HSA) for qualifying health plans
• Equity

Ideal candidate

A performance-obsessed systems engineer who treats libraries as a starting point and hardware speed as the goal. You thrive owning the full stack and maximizing throughput for ambitious reasoning systems.

Requirements

• BS in Computer Science or related technical field or equivalent industry experience
• 2+ years of relevant, hands-on industry experience
• Proficiency in Python
• Experience building or maintaining components within ML frameworks such as PyTorch, JAX, or TensorFlow
• Understanding of distributed training and collective communication primitives such as NCCL or experience deploying and profiling models on GPU-accelerated cloud infrastructure