Chegg

Senior Machine Learning / AI Engineer

Chegg
Apply
1 month ago
Madrid, SpainSenior
H1B Sponsor

Responsibilities

  • Design, develop, and deploy production-grade agentic AI systems for adaptive learning experiences.
  • Architect LLM-powered features using RAG, prompt engineering, evaluation pipelines, and multi-agent workflows.
  • Define agent orchestration, tool use, memory, and state-management patterns for production systems.
  • Move models from experimentation to production across feature engineering, training pipelines, online inference, monitoring, reliability, and quality.
  • Improve ML infrastructure and experiment orchestration tooling using platforms such as MLFlow, Airflow, SageMaker, and Kubernetes.
  • Collaborate with Data Engineers, Data Scientists, Product Managers, Designers, and engineers to embed intelligence into products and measure learning outcomes.
  • Lead research spikes covering graph-based knowledge representations, advanced RAG, fine-tuning, and agentic evaluation frameworks.

Requirements

  • Strong foundations in machine learning, applied AI, and software engineering, including clean and maintainable Python code and ML pipeline design.
  • Proven production experience building and deploying agentic AI systems with LangChain and/or LangGraph, including orchestration, tool use, and multi-step reasoning.
  • Practical knowledge of LLMs such as OpenAI, Anthropic, and HuggingFace, plus prompt engineering, vector stores, RAG architectures, and rigorous evaluation.
  • Proficiency building data and training pipelines with SQL, Airflow, or AWS services including S3, SageMaker, and Lambda.
  • Production experience with ML or AI systems, especially NLP, personalization, or recommendation systems.
  • Strong communication, collaboration, analytical thinking, ownership, and curiosity about user experience and pedagogical impact.
  • Experience with graph-based data structures or graph databases such as Neo4j or NetworkX is preferred.
  • A/B testing and impact evaluation experience is preferred.
  • Experience in EdTech, adaptive learning, or consumer personalization is preferred.

Benefits

  • Centrally located offices with free breakfast, snacks, and fresh fruit.
  • Two free lunches per week from a selection of restaurants.
  • Private health insurance.
  • Personal training budget.
  • Flexible working hours and a hybrid working model.
  • Enhanced maternity and paternity leave.
  • Frequent social activities including team lunches, Thursday socials, and quarterly events.

Tech Stack

Apache AirflowAWSKubernetesMLflowNeo4jPythonSQL
Chegg

About Chegg

1,001-5,000 employees

About Chegg: Chegg provides individualized learning support to students as they pursue their educational journeys. Available on demand 24/7 and powered by over a decade of learning insights, the Chegg platform offers students AI-powered academic support thoughtfully designed for education coupled with access to a vast network of subject matter experts who ensure quality. No matter the goal, level, or style, Chegg helps millions of students around the world learn with confidence by helping them build essential academic, life, and job skills to achieve success. Certified Great Place to Work!: http://reviews.greatplacetowork.com/chegg For More Information: https://jobs.chegg.com/ Chegg is an equal opportunity employer

Contact me