Staff Engineer, Multimodal ML & Synthetic Data

xAI; Palo Alto (CA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Comprehensive medical coverage

401(k) retirement plan

Short & long-term disability insurance

Life insurance

Job summary

A cutting-edge AI technology company in Palo Alto seeks expert engineers for its mid-training team. The role focuses on creating multimodal AI models and requires proficiency in Python, JAX, and Spark. Ideal candidates possess strong machine learning expertise and experience in large-scale data processing. The position includes a competitive salary range of $180,000 - $440,000, along with comprehensive benefits.

Qualifications

• Expertise in ML and large model scaling, with familiarity across scaling laws.
• Strong ability to design ML experiments.
• Familiarity with techniques for curating AI training data for multiple modalities.
• Strong engineering abilities in frameworks for large-scale data processing.

Responsibilities

• Design synthetic data algorithms for model intelligence.
• Distill flagship model intelligence into flash models.
• Optimize mid-training data mixtures for reinforcement learning.
• Develop evaluations for mid-training checkpoints.

Skills

Python

JAX and XLA

Spark

Ray

Job description

A cutting-edge AI technology company in Palo Alto seeks expert engineers for its mid-training team. The role focuses on creating multimodal AI models and requires proficiency in Python, JAX, and Spark. Ideal candidates possess strong machine learning expertise and experience in large-scale data processing. The position includes a competitive salary range of $180,000 - $440,000, along with comprehensive benefits.