Talent.com
Bjak
Remote Staff Machine Learning EngineerBjak • Cambridge, England
Remote Staff Machine Learning Engineer

Remote Staff Machine Learning Engineer

Bjak • Cambridge, England
30+ days ago
Job type
  • Full-time
  • Remote
Job description

About the Role

A1 is building a proactive AI chat app for everyday users to bring intelligence to conversations, errands, organising and workflows. Unlike traditional chat-based applications, our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. The system must handle multi-step reasoning, interact with external tools, and remain reliable despite non-deterministic model behavior.

As Technical Lead, Machine Learning, you own the execution layer of A1’s intelligence. You translate research direction into reliable, scalable, production-grade ML systems.

This role sits at the intersection of research, infrastructure, and product. You are responsible for making models trainable, deployable, observable, and performant under real-world constraints.

What You'll Do

  • Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference architecture, and deployment.

  • Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation.

  • Architect and operate scalable inference systems, balancing latency, cost, and reliability.

  • Design and maintain data systems for high-quality synthetic and real-world training data.

  • Implement evaluation pipelines covering performance, robustness, safety, and bias, in partnership with research leadership.

  • Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.

  • Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products.

  • Make pragmatic trade-offs and ship improvements quickly, learning from real usage.

  • Work under real production constraints: latency, cost, reliability, and safety

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.

  • You are self-directed, pragmatic, and take full ownership of outcomes.

  • You communicate clearly and collaborate well in small, high-trust teams.

How We Work

The best products today in the world were built by small, world class teams. We are a high talent density and hands-on team. We make decisions collectively, move at rapid speed, striking a balance between shipping high quality work and learning. Joining our team requires the ability to bring structure, exercise judgment, and execute independently. Our goal is to put in hands of our users a truly magical product

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.

Create a job alert for this search

Remote Staff Machine Learning Engineer • Cambridge, England

Similar jobs

Senior Machine Learning Engineer (Large Systems)

GraphcoreCambridge, England, GB
Full-time

Bristol, UK; Cambridge, UK; London, UK.Graphcore is one of the world’s leading innovators in Artificial Intelligence compute.It is developing hardware, software and systems infrastructure that will... Show more

 • Promoted

Software Engineer (Machine Learning)

ArmCambridge, England, GB
Full-time

Arm's Machine Learning group in Cambridge is hiring! Would you like to help analyse and improve the latest Machine Learning or Artificial Intelligence solutions on Arm hardware and software?.Succes... Show more

 • Promoted

Senior Machine Learning Applications and Compiler Engineer

NVIDIACambridge, England, GB
Full-time

Build, develop, and maintain high-performance runtime and compiler components, focusing on end-to-end inference optimization.Define and implement mappings of large-scale inference workloads onto NV... Show more

 • Promoted

Machine Learning Research Engineer

LuminanceCambridge, England, GB
Full-time

This is a fantastic opportunity to join market-leading UK AI company, Luminance.Named in Tech Nation’s prestigious Future Fifty list and the recipient of two Queen’s Awards, Luminance is the world’... Show more

 • Promoted

Senior Machine Learning Test Engineer United Kingdom

AutodeskCambridge, England, GB
Full-time

As a Senior Machine Learning QA Engineer in the Research Enablement team, you will work side-\u2011by-\u2011side with researchers, Machine Learning Engineers and software engineers to define and up... Show more

 • Promoted

Machine Learning Engineer

ECM Selection Ltd.Cambridge, ENG, GB
Full-time

Inventing innovative machine learning systems to solve real-world problems in the defence sector.Cambridge; to c£70,000 DoE + Benefits.This innovative team of engineers and scientists are usi... Show more

 • Promoted

Machine Learning Engineer

ECM Selection (Holdings) LimitedCambridgeshire and Peterborough, England, GB
Full-time

Inventing innovative machine learning systems to solve real-world problems in the defence sector.Cambridge; to £70,000 DoE + Benefits.This innovative team of engineers and scientists are using mach... Show more

 • Promoted

Senior Research Engineer - AI & Machine Learning

European Tech RecruitCambridge, England, GB
Full-time

We are supporting a pioneering technology organisation seeking a.AI research and engineering initiatives.This position sits at the intersection of advanced research and practical engineering.You wi... Show more

 • Promoted

Remote AI Engineer — Build Scalable ML Pipelines

YO IT ConsultingCambridge, England, GB
Remote
Full-time

YO IT Consulting is seeking an AI Engineer to contribute to the development of next-generation AI systems.This remote position requires proficiency in machine learning and the ability to build and ... Show more

 • Promoted

Senior Applied AI/ML Engineer

TetraScienceCambridge, England, United Kingdom
Full-time

Responsible for designing, developing, training, and validation of AI/ML products.Support and advise executive leadership regarding technical and commercial feasibility.Work with commercial teams t... Show more

 • Promoted

AI/ML Engineer

IC ResourcesCambridge, England, GB
Full-time

A health technology company is seeking an On-Device ML Engineer to develop machine learning models that run directly on wearable devices, extracting reliable health signals under strict real-world ... Show more

 • Promoted

Machine Learning Engineer | Neurobiology | Dr Albert Cardona | LMB 2611

UKRICambridge, England, GB
Temporary

This is a fixed term position for 2 years due to time limited funding from an external grant.To lead the deployment, scaling and maintenance of machine learning systems for petabyte-scale connectom... Show more

 • Promoted

Staff Engineer — ML Validation & Benchmarking

EngineersOfAICambridge, England, GB
Full-time

EngineersOfAI is searching for a Machine Learning Validator in Cambridge, UK.The chosen candidate will validate complex ML systems and ensure they run reliably on advanced infrastructure.The role r... Show more

 • Promoted

Machine Learning Engineer

ECM SelectionCambridge, England, GB
Full-time

This innovative team of engineers and scientists are using machine learning tightly integrated with modern electronics to create new classes of products and radically alter the shape, performance a... Show more

 • Promoted

Machine Learning Senior Software Engineer

ArmCambridge, England, GB
Full-time

Arm's Central Engineering Machine Learning Group is seeking highly motivated and creative Software Engineers to join a cross-site and multi-culture software team.This role presents an excellent opp... Show more

 • Promoted

Analytics Engineer: Lakehouse Data Models & Metrics

CMR SurgicalCambridge, England, GB
Full-time

This job is brought to you by Jobs/Redefined, the UK's leading over-50s age inclusive jobs board.We're growing fast, and we're on a mission to transform how surgery is delivered globally-for the be... Show more

 • Promoted

Staff Engineer (ML Engineer)

EngineersOfAICambridge, England, GB
Full-time

Validate the ML stack that turns accelerator hardware into trusted AI performance.This role sits where modern ML models meet Graphcore’s software and hardware stack.You will test, benchmark and val... Show more

 • Promoted

Machine Learning Research Engineer

Luminance Technologies LtdCambridge, England, GB
Full-time

This is a fantastic opportunity to join market-leading UK AI company, Luminance.Named in Tech Nation’s prestigious Future Fifty list and the recipient of two Queen’s Awards, Luminance is the world’... Show more

 • Promoted

Machine Learning Integration Engineer

Darktrace LtdCambridge, England, GB
Full-time

Machine Learning Integration EngineerApplylocations: Cambridge Office, United Kingdomtime type: Full timeposted on: Posted Todayjob requisition id: JR101685Darktrace is a global leader in A... Show more

 • Promoted

Staff Imaging Algorithms Engineer

Analog Devices, Inc.Cambridge, England, GB
Full-time

Algorithms Engineer in Cambridge, UK, to develop core algorithms for optical biosensing and imaging technologies.The role demands a strong background in algorithm development, signal processing, an... Show more