Multimodal AI Data Infra Engineer - Equity + Unlimited PTO

Avala

San Francisco (CA)

Hybrid

USD 150,000 - 230,000

Full time

12 hours ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Competitive salary
Equity
Health insurance
Dental and vision
FSA
Unlimited PTO

Job summary

Avala is building multimodal training data infrastructure in San Francisco to power Physical AI for autonomous vehicles and robotics. You will design and implement large-scale data pipelines that process petabytes of video, sensor, and action data from a global network of annotation experts.

The role emphasizes scaling models, distributed processing, and modern DL frameworks, offering competitive compensation, equity, and health benefits in a fast-moving AI hardware context.

Qualifications

  • 3+ years experience scaling and optimizing large AI models.
  • Proficiency in Python with strong software engineering practices.
  • Deep learning fundamentals: optimization, loss functions, neural network architectures.
  • Experience with PyTorch, TensorFlow, or JAX.
  • Strong expertise in distributed computing and parallel processing.
  • Experience with petabyte-scale data processing pipelines.

Responsibilities

  • Create multimodal training datasets from diverse robotics and AV data sources.
  • Develop infrastructure for training large multimodal models at scale.
  • Design data processing pipelines for video, LiDAR, radar, and sensor fusion.
  • Build downstream evaluation systems to guide model tuning and dataset composition.
  • Run scaling experiments for model size, dataset quality, and training efficiency.

Skills

Python
Distributed computing
Deep learning
Model scaling
Petabyte-scale data

Tools

PyTorch
TensorFlow
JAX
Data pipelines

Job description

Avala is building multimodal training data infrastructure in San Francisco to power Physical AI for autonomous vehicles and robotics. You will design and implement large-scale data pipelines that process petabytes of video, sensor, and action data from a global network of annotation experts.

The role emphasizes scaling models, distributed processing, and modern DL frameworks, offering competitive compensation, equity, and health benefits in a fast-moving AI hardware context.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

ML Infra Engineer: Petabyte-Scale Training & Data Pipelines
ML Infra Engineer: Petabyte-Scale Training & Data Pipelines

Avala • San Francisco (CA)

Hybrid
USD 170,000 - 250,000
Equity
Health insurance
Dental insurance
+3
Multimodal Data Engineer: End-to-End Pipelines + Equity
Multimodal Data Engineer: End-to-End Pipelines + Equity

ABAKA AI • Mountain View (CA)

On-site
USD 120,000 - 225,000
Equity in Abaka AI
Comprehensive benefits: health, dental
Vision coverage
+1
AI Engineer, Training Data Infrastructure
AI Engineer, Training Data Infrastructure

Avala • San Francisco (CA)

Hybrid
USD 150,000 - 230,000
Competitive salary
Equity
Health insurance
+3
Multimodal AI Engineer: Data Pipelines & Evaluation
Multimodal AI Engineer: Data Pipelines & Evaluation

Meta • United States

Remote
USD 154,000 - 217,000
Bonus
Equity
Benefits
AI Engineer, Data Scaling & Optimization
AI Engineer, Data Scaling & Optimization

Avala • San Francisco (CA)

Hybrid
USD 170,000 - 250,000
Equity
Health insurance
Dental insurance
+3
Senior Data Platform Engineer - Fleet AI Infra (Equity)
Senior Data Platform Engineer - Fleet AI Infra (Equity)

Avala • San Francisco (CA)

Hybrid
USD 170,000 - 220,000
Health, dental, vision coverage
FSA
Unlimited PTO
+1
Multimodal ML Engineer — Early-Stage, Equity & PTO
Multimodal ML Engineer — Early-Stage, Equity & PTO

Abaka AI • Palo Alto (CA)

On-site
USD 175,000 - 275,000
Equity
Comprehensive benefits package (health, dental, vision)
Flexible work schedule
+1
Senior ML Engineer - 3D Perception & Tracking Unlimited PTO
Senior ML Engineer - 3D Perception & Tracking Unlimited PTO

Avala • San Francisco (CA)

Hybrid
USD 150,000 - 210,000
Competitive salary
Equity
Comprehensive health/dental/vision
+2
Senior Data Engineer - Fleet Sensor Data & AI Infra
Senior Data Engineer - Fleet Sensor Data & AI Infra

Avala • San Francisco (CA)

Hybrid
USD 130,000 - 190,000
Health care (medical, dental, vision)
FSAs
Unlimited PTO
+1
ML Engineer, Agentic Labeling
ML Engineer, Agentic Labeling

Avala • San Francisco (CA)

Hybrid
USD 150,000 - 230,000
Equity
Health insurance
Unlimited PTO
+1