Skip to content
RipplingRipplingSan Francisco, CA

Staff Software Engineer

Build and scale AI platforms for Rippling's Data Cloud, focusing on schema retrieval, query planning, LLM training pipelines, RL environments, and agent harnesses for intelligent workforce workflows. Requires 8+ years experience with production LLMs, distributed systems, and cloud infrastructure.

189k – 315k/yr
Hybrid8+ YOEML Engineering

About the role

What you will do

  • Develop a state of the art schema retrieval system that operates over Rippling native and customer defined schema.
  • Implement and scale Data Cloud’s AI training pipelines for data retrieval, from designing data generation to managing training clusters.
  • Design and build RL training environments for structured and unstructured data retrieval.
  • Develop agent harnesses enabling product teams to safely build AI features on Data Cloud.
  • Optimize model serving and inference for scale from GPU level performance to agentic user experiences.
  • Partner with AI Platform, Infrastructure, Security, and Product Engineering to drive architecture, standards, and rollout plans.
  • Lead technical direction, mentor engineers, and drive execution on multi-team, ambiguous initiatives.

What you will need

  • 8+ years of software engineering experience, including significant ownership of distributed systems in production.
  • Experience post training and deploying LLMs in production environments.
  • Experience optimizing model inference at scale, particularly with LLMs and embedding models.
  • Strong backend engineering skills in one or more languages such as Python, Go, or Java.
  • Experience with cloud-native infrastructure (Kubernetes, container orchestration, observability, reliability engineering).
  • Ability to drive cross-functional technical strategy and execute through influence across teams.

Skills

LLMsPythonGoJavaKubernetesDistributed Systemsmodel inferencerl trainingschema retrievalagent harnesses

Similar roles

ML Engineering jobs
Rippling

Staff Software Engineer

RipplingNew York, NY

Build foundational AI agent infrastructure at Rippling, owning agent creation, invocation, skill packaging, plugin connectivity, and automations. Lead platform and distributed systems work with 8+ years experience, technical leadership, and cross-functional impact.

189k – 315k/yrHybrid8+ YOEML Engineering
Scale AI

Staff Software Engineer, Public Sector

Scale AISan Francisco, CA +3

Leads development of agentic AI systems for public sector, including guardrails, data processing, and fleet orchestration for federal datasets. Mentors engineers, defines technical strategy, and communicates with stakeholders to ensure reliable, secure solutions.

189k – 362k/yrOn-siteML Engineering
Pinterest

Staff Machine Learning Engineer, Applied Research

PinterestSan Francisco, CA

Staff Machine Learning Engineer conducting applied research on multi-modal LLMs, agents, NLP, computer vision, graph neural networks, and recommender systems to improve Pinterest search, recommendations, and discovery. Requires MS/PhD and 6+ years industry experience building production ML models.

189k – 390k/yrHybrid7+ YOEML Engineering
Pinterest

Staff Machine Learning Engineer, Computer Vision

PinterestSan Francisco, CA

Staff Machine Learning Engineer developing state-of-the-art visual encoders and multimodal models at Pinterest Labs. Prototype visual reasoning tools, train billion-scale models on rich visual-text data, ship to production for recommender systems and VLMs, publish research, and mentor juniors. Requires strong CV/ML background, publications, and PhD or equivalent.

189k – 390k/yrHybrid7+ YOEML Engineering
Databricks

Staff Software Engineer

DatabricksSan Francisco, CA

Build and optimize LLM inference infrastructure at enterprise scale for partner and self-hosted frontier models. Requires 8+ years backend/infrastructure engineering experience with distributed systems, real-time serving, and ML/GPU orchestration.

190k – 265k/yrOn-site8+ YOEML Engineering