Sr. Staff Engineer, AI Platform
About the Team:
The AI & Data Platform team builds the foundation that powers applied AI across Quizlet: personalization and recommendations, retrieval and ranking, AI Coach, generative content, and emerging agentic experiences. We own the systems that make model development fast, reliable, observable, and safe, from data and features through training, evaluation, deployment, and inference.
This team is pragmatic about build versus buy. We aggressively use the right mix of managed Google Cloud services, best-in-class vendor tooling, open-source infrastructure, and internal platform abstractions when that gives Quizlet the best combination of speed, reliability, and leverage.
About the Role:
As a Senior Staff Engineer on the AI Platform team, you will define the technical direction for Quizlet’s next generation of ML and LLM infrastructure. This is a deeply hands-on, org-level individual contributor role. You will architect critical platform systems, drive build-versus-buy decisions, partner with leaders across Applied AI, Data Science, Product Engineering, and Infrastructure, and raise the bar for how models and LLM-powered systems are trained, evaluated, shipped, served, and governed across the company.
This role is ideal for an engineer who can operate at senior-staff scope in a large company, but wants the speed, ownership, and breadth of impact that come with a smaller, cloud-native environment. At Quizlet, the role spans the real stack rather than a narrow subsystem: Google Cloud, Kubernetes and GKE, distributed training, MLflow-centered workflows, data and feature foundations, online and asynchronous inference, and the evaluation and observability needed to run predictive ML and LLM systems safely at scale.
We’re happy to share that this is an onsite position in our San Francisco office. To help foster team collaboration, we require that employees be in the office a minimum of three days per week: Monday, Wednesday, and Thursday and as needed by your manager or the company. We believe this work environment enhances efficiency, fosters collaboration, and supports growth for both employees and the organization.
In this role, you will:
- Set the multi-year architecture and technical roadmap for Quizlet’s AI platform across data, features, model development, evaluation, deployment, and serving
- Standardize MLflow-based workflows for experiment tracking, model packaging, artifact lineage, model registry, promotion, rollback, and inference deployment patterns
Want more jobs like this?
Get Data and Analytics jobs in San Francisco, CA delivered to your inbox every week.
What Success Looks Like in 12 Months:
What you bring to the table:
- 10+ years of experience building large-scale ML, data, or inference platforms in production environments
- Proven track record operating as a staff or senior-staff IC, setting technical direction across multiple teams or an entire engineering organization
- Deep expertise in distributed systems and cloud infrastructure, preferably on Google Cloud, including containerized workloads on Kubernetes and GKE
- Strong familiarity with MLflow for model training and inference workflows, including experiment tracking, reproducibility, model registry, packaging, promotion, and deployment patterns
- Experience building platforms for model training, feature management, dataset lineage, online and batch inference, and model observability
- Hands-on experience with LLM and GenAI infrastructure, such as model serving, retrieval-augmented generation, vector retrieval, evaluation frameworks, prompt and version management, and safety or quality guardrails
- Strong engineering skills in Python plus one or more backend or systems languages such as Go, Java, or Scala
- Strong judgment in system design, reliability, security, privacy, and cost-performance tradeoffs
- Exceptional communication and stakeholder leadership skills, with a record of mentoring senior engineers and driving alignment in ambiguous environments
Bonus points if you have:
- Experience with Ray, Vertex AI, BigQuery, Pub/Sub, Spark, Flink, dbt, feature stores, Triton, vLLM, or similar platform technologies
- Experience supporting ranking, retrieval, search, recommendation, personalization, or other consumer-facing ML systems
- Experience building evaluation and observability tooling for both predictive models and LLM-based systems
- Experience in EdTech, consumer learning products, or domains where trust, quality, and safety matter deeply
Compensation, Benefits & Perks:
- Quizlet is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. Salary transparency helps to mitigate unfair hiring practices when it comes to discrimination and pay gaps. Total compensation for this role is market competitive, including a starting base salary of $245,000 - $320,000, depending on location and experience, as well as company stock options
- Collaborate with your manager and team to create a healthy work-life balance
- 20 vacation days that we expect you to take!
- Competitive health, dental, and vision insurance (100% employee and 75% dependent PPO, Dental, VSP Choice)
- Employer-sponsored 401k plan with company match
- Access to LinkedIn Learning and other resources to support professional growth
- Paid Family Leave, FSA, HSA, Commuter benefits, and Wellness benefits
- 40 hours of annual paid time off to participate in volunteer programs of choice
Perks and Benefits
Health and Wellness
Parental Benefits
Work Flexibility
Office Life and Perks
Vacation and Time Off
Financial and Retirement
Professional Development
Diversity and Inclusion