Senior Machine Learning / AI Engineer

Chegg, Inc.; Santa Clara (CA); Hybrid; USD 150,000 - 230,000; Full time

Benefits offered by this job

Free breakfast & snacks

2 free lunches per week

Private health insurance

Job summary

Busuu, part of the Chegg family, is seeking a Senior Machine Learning / AI Engineer to build agentic AI systems powering adaptive learning experiences. You will own end-to-end ML workflows, from data pipelines to production inference, with emphasis on LLMs, RAG, and multi-step reasoning in production environments.

You will collaborate with Data Scientists, Product Managers, and Designers to deliver impact in an EdTech setting while leveraging LangChain/LangGraph and modern ML tooling.

Qualifications

• Strong foundations in machine learning, applied AI, and software engineering.
• Experience building and deploying agentic AI systems in production.

Responsibilities

• Build & scale agentic AI systems powering adaptive learning experiences.
• Architect and integrate LLM-powered features with RAG and evaluation pipelines.
• Lead multi-agent orchestration using LangChain/LangGraph and define tooling patterns.
• Own full ML lifecycle from experimentation to production, including data pipelines and monitoring.
• Collaborate with data scientists, product managers, designers and engineers to embed intelligence into products.

Skills

Python

ML pipelines

LangChain

LangGraph

LLMs

RAG architectures

prompt engineering

vector stores

A/B testing

SageMaker

Tools

MLFlow

Airflow

Kubernetes

OpenAI

Anthropic

HuggingFace

Job description

Job DescriptionAbout usJoining Busuu means being part of one of the top EdTech companies in the world, a multiple award-winner recognised for its innovation and impact in language learning.Busuu's vision is to empower people through languages. We are the world's largest online community for language learning, with 120+ million registered users. We make learning a language easy by combining AI-powered courses with feedback from our global community of native speakers and lesson content designed for real life.Busuu is part of the global Chegg family. Chegg is the leading student-first connected learning platform and a NYSE listed company.What does a Senior Machine Learning / AI Engineer do at Busuu?Build & scale agentic AI systems: Own the design, development and deployment of production-grade agentic systems that power adaptive learning experiences — from multi-step reasoning pipelines to autonomous feedback loops that respond to learner behaviour in real time.LLMs & RAG architectures: Architect and integrate LLM-powered features using retrieval-augmented generation (RAG), prompt engineering strategies, and evaluation pipelines. Lead application of these to high-impact use cases such as mistake analysis, content generation, and personalised learning paths.Agentic frameworks: Lead the design of multi-agent workflows using frameworks such as LangChain and LangGraph. Define agent orchestration patterns, tool use, memory, and state management strategies for production environments, and establish best practices across the team.Full ML lifecycle ownership: Collaborate with Data Scientists and Senior ML Engineers to move models from experimentation to production, including feature engineering, training pipelines, online inference, and monitoring. Take ownership of reliability and quality end to end.Platform & tooling development: Drive improvements to our ML infrastructure and experiment orchestration tools (e.g. MLFlow, Airflow, SageMaker, Kubernetes), and help make AI development faster and safer across the team.Cross-functional collaboration: Work closely with Data Engineers, Product Managers, Designers, and other engineers to embed intelligence into our products, improve experimentation velocity, and drive measurable learning outcomes.Research & innovation: Lead research spikes on emerging AI/ML technologies — from graph-based knowledge representations to advanced RAG patterns, fine-tuning strategies, and agentic evaluation frameworks. Shape our evolving AI strategy and contribute to the wider engineering community.What we're looking for in a Senior Machine Learning / AI EngineerStrong foundations in machine learning, applied AI, and software engineering. You write clean, maintainable Python code, are comfortable designing ML pipelines, and can work across APIs and microservices at scale.Proven experience building and deploying agentic AI systems using LangChain and/or LangGraph — including agent orchestration, tool use, and multi-step reasoning pipelines in production environments.Deep practical knowledge of LLMs (e.g. OpenAI, Anthropic, HuggingFace) and hands-on experience with prompt engineering, vector stores, and RAG architectures. You know how to evaluate and iterate on these systems rigorously.Proficiency in building data and training pipelines using tools like SQL, Airflow, or AWS services (S3, SageMaker, Lambda).Proven track record deploying ML or AI systems to production, especially around NLP, personalisation, or recommendation. A/B testing and impact evaluation experience is a strong plus.Graph-based data structures or graph databases (e.g. Neo4j, NetworkX) is a strong plus.Excellent communication skills and ability to work collaboratively in a diverse, cross-functional team. You take ownership, iterate fast, and make others around you better.Strong analytical thinking and genuine curiosity about the user experience and pedagogical impact of AI solutions.Experience in EdTech, adaptive learning, or consumer personalisation is a big plus, not a requirement.At Busuu we want to ensure that you have access to some great benefitsCentrally located offices with free breakfast, snacks, and fresh fruit2 free lunches per week from a wide selection of restaurantsGreat Private Health Insurance schemePersonal training budget to keep growingFlexible working hours and a hybrid model of workingEnhanced maternity and paternity leaveFrequent social activities: team lunches, Thursday socials, quarterly eventsWhat happens next?CV review – We'll review your application as quickly as possible.Let's chat – A quick call with our team about your experience and the role.First Interview – With the Hiring Manager.Cultural Interview – Interview about soft skills and value alignment with the team.Technical Interview – Structured questions covering core AI/ML engineering knowledge with team members.Our platform is for everyone, and so is our workplace. We embrace our differences — cultural, racial, religious, or otherwise - and believe every voice matters.If this sounds like the kind of place where you coulxd thrive, we'd love to hear from you.About UsWhat is Chegg?An ‘always on’ digital learning platform.Chegg puts students first…Everything we build in this company is student-focused, making us the leading student-first connected learning platform. Chegg strives to improve the overall return on investment in education by helping students learn more in less time and at a lower cost. This is achieved by providing students a multitude of educational tools from affordable textbook rentals to Chegg Study which supplements their learning through 24/7 tutor access, step-by-step help with questions, and more.Chegg is a publicly-held company based in Santa Clara, California and trades on the NYSE under the symbol CHGG.