Staff Software Engineer - AI jobs in United States
cer-icon
Apply on Employer Site
company-logo

Rippling · 2 hours ago

Staff Software Engineer - AI

Rippling is a company that provides integrated HR, IT, and Finance solutions. They are seeking a Staff AI/ML Engineer to architect and lead the development of production-grade AI systems, focusing on recommendation engines and ML pipelines, while also mentoring engineers and solving complex challenges in production AI systems.

EmploymentHuman Resources
check
H1B Sponsor Likelynote

Responsibilities

Architect, build, and optimize recommendation engines, personalization systems, and classification models for GTM automation
Design and implement multi-LLM architectures combining OpenAI, Claude, and Databricks models for intelligent decisioning and reasoning
Build, train, and evaluate models
Deploy and serve models using FastAPI, Kubernetes, and async microservices, with observability built in
Develop MLOps workflows for fine-tuning, retraining, model versioning, and automated evaluation
Design medallion data architectures (Bronze/Silver/Gold) using Databricks Delta Live Tables and CDC patterns
Build real-time and batch data pipelines leveraging Kafka and Databricks for high-volume model inputs
Develop and maintain embedding systems and matrix factorization-based recommendation frameworks for personalization and ranking
Implement AI data quality and monitoring frameworks to ensure reliability and trust in model outputs
Implement AI observability (LangSmith, Braintrust) to track performance, bias, and drift
Build fallback and routing systems for multi-model deployments
Optimize cost and latency through batching, caching, and adaptive model selection
Lead design reviews and guide architecture for AI/ML-driven systems
Mentor engineers on LLM integration, MLOps, and recommendation systems
Collaborate closely with product and GTM partners to translate business goals into AI-driven automation

Qualification

AI/ML ArchitectureRecommendation EnginesMLOps WorkflowsData EngineeringMulti-LLM IntegrationDatabricksKubernetesKafkaSoft Skills

Required

7+ years of software engineering experience, including 3+ years building production ML systems
Expertise in recommendation engines, matrix factorization, and personalization models
Deep experience integrating LLMs (OpenAI, Claude, etc.) into production applications
Hands-on experience training, evaluating, and deploying models in Databricks notebooks and Spark pipelines
Experience with MLOps tooling for off-the-shelf models like XGBoost, CatBoost, or LightGBM
Strong background in data engineering (Kafka, Spark, Databricks, PostgreSQL)
Proven ability to architect scalable AI systems and lead end-to-end deployment

Preferred

Familiarity with LangChain, LangSmith, and vector databases
Deep understanding of multi-LLM coordination patterns, dynamic prompt routing, and evaluation loops
Experience implementing AI safety, guardrails, and interpretability frameworks
Experience deploying containerized AI services on Kubernetes
Solid understanding of feature stores, experiment tracking, and online/offline evaluation

Benefits

Benefits
Equity

Company

Rippling

company-logo
Rippling is a workforce management platform that unifies HR, IT, and finance operations into a single system.

H1B Sponsorship

Rippling has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
2024 (42)
2023 (70)
2022 (81)
2021 (33)
2020 (6)

Funding

Current Stage
Late Stage
Total Funding
$2.44B
Key Investors
CoatueGreenoaksFounders Fund
2025-05-09Series G· $450M
2024-04-22Series F· $200M
2024-04-22Secondary Market· $590M

Leadership Team

leader-logo
Albert Strasheim
Chief Technology Officer
linkedin
leader-logo
Adil Syed
CFO
linkedin
Company data provided by crunchbase