BJAK Logo

BJAK

Lead Engineer, Machine Learning

Reposted 5 Hours Ago
Be an Early Applicant
Remote
Hiring Remotely in China
Entry level
Remote
Hiring Remotely in China
Entry level
Leads development of production machine learning systems across data, training, fine-tuning, evaluation, inference, deployment, monitoring, and optimization. Builds scalable large-model pipelines, high-performance GPU infrastructure, and reliable production systems while partnering with research and application engineering. Owns technical execution, performance measurement, debugging, system reliability, and translating model capabilities into product improvements.
The summary above was generated by AI
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.

Similar Jobs

4 Days Ago
In-Office or Remote
Expert/Leader
Expert/Leader
Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
Lead Ericsson Antenna System’s Asia business, owning regional strategy, sales, business development, customer partnerships, and profit and loss. The role manages antenna and filter product lines, establishes strategic customer relationships, collaborates with market-area organizations, and directs annual business planning and target setting. Success requires telecommunications expertise, international leadership, large-account management, customer-facing experience, strong business acumen, and the ability to lead culturally diverse teams.
Top Skills: Ericsson Antenna SystemMobile Communication AntennasRf FiltersTelecommunications
Senior level
Artificial Intelligence • Fintech • Payments • Business Intelligence • Financial Services • Generative AI
Leads APAC merchant risk operations for payments, owning team performance, service levels, quality, fraud and chargeback metrics, investigations, monitoring, and escalation workflows. Partners with Legal, Risk, Compliance, Product, and Engineering to implement controls and improve risk tooling. Drives automation and AI adoption, manages senior staff across locations, influences stakeholders, and ensures portfolio risk aligns with enterprise risk appetite.
Top Skills: AIAi/MlAnalytics DashboardsCase Management SystemsData ToolsLlmsRules Engines
8 Days Ago
Remote
Entry level
Entry level
Artificial Intelligence • Healthtech • Machine Learning • Natural Language Processing • Biotech • Pharmaceutical
Supports pharmaceutical product academic promotion and professional customer communication within an assigned territory. Responsibilities include visiting customers, communicating and gathering feedback on drug information, organizing product academic promotion activities, and meeting professional, behavioral, compliance, and performance objectives under regional management guidance.

What you need to know about the Bristol Tech Scene

Along with Gloucester, Swindon and Bath, Bristol is part of the "Silicon Gorge" tech hub, a region in the U.K. renowned for its high-tech and research-driven industries, with a particular emphasis on sustainability and reducing environmental impact. As the European Green Capital, Bristol is home to 25,000 cleantech companies, including Baker Hughes and unicorn Ovo Energy. The city has committed to achieving net-zero emissions within the next decade.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account