Job Details

Get AI-powered advice on this job and more exclusive features.

Stealth provided pay range

This range is provided by Stealth. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$150,000.00/yr - $300,000.00/yr

Bay Area (Hybrid)

Can you build & optimize distributed ML pipelines with Ray or Spark?

Do you love speeding up cloud infra (Kubernetes, Docker, CI/CD)?

Excited to build the data backbone for large-scale ML training?

We're a tier 1 VC-backed start-up, developing hyper-realistic 3D simulations using AI. Our customers include leading names in industries such as autonomous vehicles, drones and robotics.

Role

You'll be hands‑on improving CI/CD pipelines, speeding up Docker builds, and scaling scene processing on Ray. You'll also:

• Build high-performance data pipelines for multimodal datasets (3D, video, sensor).
• Optimize distributed training and processing across Spark, Databricks, and Kubernetes.
• Work with researchers to productionize PyTorch models and streamline ML workflows.
• Develop tools that make data discoverable, reusable, and reliable throughout the ML lifecycle.

You

• Strong Python skills and experience with distributed systems (Ray, Spark, Flyte, Dask).
• Hands‑on with cloud, Kubernetes, and distributed training (Ray, PyTorch DDP, Horovod).
• Familiar with dataset versioning and experiment tracking (DVC, MLflow).

Bonus Points

• Experience in simulation, robotics, or autonomy pipelines.
• Background in deep learning (PyTorch) and 3D / sensor data (LIDAR, meshes, radiance fields).
• Open-source contributions or frontend/UI experience.

Seniority level

• Mid-Senior level

Employment type

• Full-time

Job function

• Software Development

Referrals increase your chances of interviewing at Stealth by 2x