Bjak brand banner

Lead Engineer, Machine Learning

BjakUnited KingdomPosted 2h agoonsite
via Arbeitnow

About the Role

There 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 hallucinations

Our objective is to organise anyone's life, allowing us all to spend time on valuable and meaningful things

As 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 Own

  • Own 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 For

  • Experience 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.

 

Outcomes

  • Research 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 Stack

  • Python

  • PyTorch / JAX

  • GPU-based training and inference system

 

Ideal Experience

  • You 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 Work

We 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 process

If 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.

Find more English Speaking Jobs in United Kingdom on Arbeitnow

Skills

  • A1 Engineering

Similar roles

  • Zhejiang Dingli Machinery Co.,ltd logo

    Parts Advisor

    Zhejiang Dingli Machinery Co.,ltd·West Bromwich, United Kingdom

    Responsibilities * Order Prep:Pre-pick workshop parts, pick customer orders, and check stock availability per enquiry. * Sales & Customer Service:Handle inbound/outbound calls and walk-ins; build relationships via product knowledge and strong first impressions. * Stock Control:Replenish consumables/service parts regularly for quick service to workshop, departments, and trade/retail clients. * Process warranty claims (parts-related) and complete admin (customer records, invoicing) promptly…

    • Full-time
  • dept logo

    Senior Account Manager (Freelance)

    dept·Manchester, United Kingdom

    <p><em>Please submit your application in English.</em></p> <p><em>Must be fluent, C1/C2 or native in German and English languages.</em></p> <p><strong><em>Please note: This is a short-term freelance contract opportunity.</em></strong></p> <p><em>DEPT® is a Growth Invention company built to help the world’s most ambitious brands grow faster. Operating at the intersection of technology and marketing, our 4,000+ specialists deliver growth invention services across Brand &amp;…

    • On-site
    • Full-time
  • Beamery logo

    Senior IT Manager, Contractor or FTC

    Beamery·London, United Kingdom

    ABOUT BEAMERY With the rapid expansion of AI and automation, the future of work has never been more challenging for organizations. Beamery’s unique jobs, skills and tasks data platform helps organizations navigate these challenges and make more informed decisions across Talent Lifecycle Management. Our solutions power recruitment, mobility, upskilling, diversity, work architecture and workforce planning for some of the world’s…

    • On-site
    • Full-time
  • Primer.io logo

    Product Marketing Manager

    Primer.io·United Kingdom

    An Introduction to Primer Primer is the unified infrastructure for global payments. We give finance and payments teams the visibility and control to reduce complexity, improve performance, and capture more revenue - all from a single platform. Backed by Sofina, Peak XV Partners, ICONIQ, Tencent, Accel, and Balderton, we're building the payments layer the world's best companies rely on. Watch…

    • On-site
    • Full-time
  • Bjak logo

    Organizational Effectiveness Manager

    Bjak·United Kingdom

    About A1 There are over 5 billion users using basic applications today such email, notes, tasks that are not AI-native. Our mission is to build a proactive smart assistant for everyday users to bring intelligence to conversations, errands, organising and workflows, with minimal prompting. Our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion.…

    • On-site
    • Full-time
  • Bjak logo

    Technical Product Lead, AI Email App

    Bjak·United Kingdom

    ABOUT THE ROLE There 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-site
    • Full-time
  • Bjak logo

    Product Manager - AI Finance

    Bjak·United Kingdom

    About BJAK The original mission of BJAK is we believe people deserve smarter ways to plan, save and grow their money. This is the origin of our name. Started in 2019, we built the first mobile-first, insurance platform, enabling insurance to be accessible online by millions in the region. Today, its the leading insurance platform in Southeast Asia. Today, we…

    • On-site
    • Full-time
  • Bjak logo

    Principal Software Engineer, AI Email App

    Bjak·United Kingdom

    ABOUT THE ROLE There 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-site
    • Full-time