Staff Software Engineer, AI Platform
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
Lead the development of tooling and datasets that accelerate AI model training and experimentation.
Design and implement scalable data pipelines to power continuous improvement via human feedback.
Own data infrastructure tasks spanning Spark, Beam, and GCP, ensuring reliability and performance.
Collaborate with ML researchers to translate data insights into improved models and evaluation metrics.
Build and maintain data warehousing, pipelines, and tooling with BigQuery, dbt, Ray, Beam, and Python/SQL.
Advise on containerization, cloud architecture, and governance to support rapid growth and scale.
Design and implement scalable data pipelines to power continuous improvement via human feedback.
Own data infrastructure tasks spanning Spark, Beam, and GCP, ensuring reliability and performance.
Collaborate with ML researchers to translate data insights into improved models and evaluation metrics.
Build and maintain data warehousing, pipelines, and tooling with BigQuery, dbt, Ray, Beam, and Python/SQL.
Advise on containerization, cloud architecture, and governance to support rapid growth and scale.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.
? Staff Software Engineer, Agentic AI Systems ? Staff Software Engineer, Agentic AI Systems ? Staff Software Engineer, AI Authoring ? Staff Software Engineer - Front End AI (Bangkok based, Relocation provided) ? Staff Software Engineer - Full Stack AI (Bangkok based, Relocation provided) ? Staff Software Engineer (BE/ML) - Compliance & Readiness
Remote workno
CityRedwood City, US