Lead Engineer, Machine Learning
bjakcareer · China
About The Role
About the RoleThere are over 5 billion users using basic applications today such email, notes, tasks, calendar and they're not AI-native. Our mission is to build proactive applications for anyone in the world, who are not used to complex prompting. We aim to bring intelligence to conversations, errands, organising and workflows, with minimal to no prompting.Our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. We believe products will greatly reduce hallucinationsOur objective is to organise anyone's life, allowing us all to spend time on valuable and meaningful thingsAs Lead Engineer, Machine Learning, you own the execution layer of our intelligence, turning research and model capabilities into reliable, scalable production <systems.You> will work across the model lifecycle: data, training, evaluation, inference, and deployment. This is a hands-on leadership role for someone who wants to operate at the intersection of research, systems, and product. What You'll OwnOwn the end-to-end ML systems powering our company, from data and training to evaluation, inference, and <deployment.Build> and evolve training and fine-tuning pipelines for large <models.Design> evaluation systems that measure capability, robustness, safety, and real-world product performance.Architect high-performance inference systems, optimizing latency, GPU utilization, memory, cost, and <reliability.Build> data pipelines and systems for high-quality real-world and synthetic training data.Establish reliable production infrastructure for deploying, monitoring, and continuously improving models.Partner closely with research and application engineering to turn model capabilities into product improvements.Make pragmatic technical trade-offs and rapidly iterate based on real-world performance. What We're Looking ForExperience building and shipping ML systems used in production, not just research prototypes.Strong understanding of modern large-model training, fine-tuning, evaluation, and inference.Strong software engineering and systems fundamentals.Experience operating ML workloads at meaningful scale, particularly GPU-based systems.Strong technical judgment and the ability to navigate ambiguous problems independently.A bias toward experimentation, measurement, and shipping.High standards for correctness, reliability, and production quality. OutcomesResearch and models reliably translate into production-ready solutions with clear performance and quality <targets.ML> pipelines, training loops, and inference systems are stable, efficient, and maintainable.Production issues are detected, debugged, and resolved quickly, minimizing user <impact.Team> members are supported, aligned, and able to deliver high-impact ML work with minimal friction.Iterations on models and systems are measurable, safe, and improve user experience over time. Tech StackPythonPyTorch / JAXGPU-based training and inference system Ideal ExperienceYou have built or shipped real ML systems used by people, not just <demos.You> are comfortable working with large models and understanding their failure <modes.You> write strong, production-grade code and care about system correctness. How We WorkWe are a small, high-talent-density, hands-on team. Engineers have broad ownership and are expected to exercise strong judgment and execute independently.We make decisions quickly, work closely together, and balance speed with engineering fundamentals. We care less about process and more about building something exceptional. Interview processIf there appears to be a fit, we'll reach to schedule 3, but no more than 4 interviews.Applications are evaluated by our technical team members. Interviews will be conducted via virtual meetings and/or onsite.We value transparency and efficiency, so expect a prompt decision. If you've demonstrated the exceptional skills and mindset we're looking for, we'll extend an offer to join us. This isn't just a job offer; it's an invitation to be part of a team that's bringing AI to have practical benefits to billions globally.
Similar roles you might like
See all →This is an external listing. JobSpring does not represent or verify the employer. Report this listing
