← All roles

ML Platform Engineer

UniversalAGI· Technical Staff
Apply on company site
San FranciscoFull-timePosted 3w ago

About the role

ML Platform Engineer responsible for building and owning the execution platform powering research and customer deployments, including data generation, simulation orchestration, training/fine-tuning infrastructure, benchmarking pipelines, and production deployments in customer environments.

Responsibilities

  • Build and operate scalable infrastructure for data generation and simulation workflows (job orchestration, scheduling, queues, retries, observability).
  • Build reproducible pipelines for training/fine-tuning and benchmarking (artifact/version management, experiment tracking, dataset lineage).
  • Own cost/performance tradeoffs across compute, storage, networking, and runtime efficiency.
  • Lead deployments of our stack into customer cloud/on-prem environments (AWS/GCP/Azure + hybrid), including secure networking, permissions, and data movement.
  • Build robust deployment patterns: environment provisioning, CI/CD, rollbacks, monitoring, and incident response.
  • Partner with customers to ensure reliability and repeatability under real-world constraints (security, compliance, infra limits, data governance).

Qualifications

  • Strong software engineering skills (clean code, debugging, reliability, reproducibility).
  • Hands-on experience building/operating infrastructure for ML/compute-heavy workflows: pipelines, job orchestration, GPU compute, storage, CI/CD, monitoring.
  • Olympic athlete mindset: high standards for measurable customer metrics.
  • Resourcefulness: balance quick fixes with robust solutions and justify tradeoffs by impact.
  • Ownership: comfortable owning work end-to-end and accountable for measurable outcomes.

Skills

Required:Software engineeringML/infrastructure for data generation and simulation
Nice to have:Orchestration/workflow tooling

Benefits

  • Competitive compensation and equity.
  • Competitive health, dental, vision benefits paid by the company.
  • 401(k) plan offering.
  • Flexible vacation.
  • Team Building & Fun Activities.
  • Great scope, ownership and impact.
  • AI tools stipend.
  • Monthly commute stipend.
  • Monthly wellness / fitness stipend.
  • Daily office lunch & dinner covered by the company.
  • Immigration support.
Last updated Sep 30, 2026. Interested? Apply on the company site.
UniversalAGI logo

About UniversalAGI

Foundation models for automating physical systems engineering

The platform for engineering teams building and operating Physics AI at scale. Predict aerodynamic performance, structural loads, and thermal behavior in seconds.

PlatformAI simulationUncertainty QuantificationLIFT Volume TokenizationCalibrated UncertaintySUV-PTAerospaceDefense
Company profileWebsite

More open roles at UniversalAGI