Embodied AI / VLA Research Engineer

Santa Clara, CA, US Mid Level Research Engineer

Interested in this Research Engineer role at MaxInsights?

Apply Now →

Skills & Technologies

PythonPytorch

About This Role

AI job market dashboard showing open roles by category

Job Description:

Position Overview

In this role, you will work on the research, training, optimization, and real-world deployment of embodied AI models, including Vision-Language-Action (VLA) models, World Models, and related robotics foundation models.

You will work across multimodal perception, robot control, long-horizon task planning, large-scale robot data, and model deployment. This is a highly hands-on role that involves both model development and real-world robotic system integration.

Key Responsibilities

Embodied AI Model Development

  • Research and develop Vision-Language-Action (VLA), World Model, and other embodied AI foundation models.
  • Work on areas including:

+ Robotic manipulation + Multimodal perception and control + Long-horizon task planning + Vision-language-action reasoning + Robot-environment interaction

  • Develop and optimize models for real-world robotic applications.
  • Translate research ideas into practical models and systems that can operate reliably on physical robots.

Foundation Model Training & Optimization

  • Train and optimize embodied AI foundation models using large-scale real-robot datasets and egocentric human demonstration data.
  • Design approaches for incorporating multimodal inputs such as:

+ Vision + Force + Tactile sensing + Proprioception + Other robot and environmental signals

  • Develop and evaluate multimodal fusion architectures.
  • Conduct model training, evaluation, benchmarking, and performance optimization.
  • Analyze model performance and identify opportunities to improve training efficiency, generalization, and real-world performance.

Robotics Data Pipeline

  • Build and improve large-scale robotics data pipelines for model training.
  • Develop processes for:

+ Data cleaning + Data filtering + Resampling + Data augmentation + Data quality evaluation

  • Design scalable data pipelines capable of supporting large volumes of robot and human demonstration data.
  • Work closely with data and robotics teams to improve dataset quality and training efficiency.

Model Deployment & Real-Robot Testing

  • Deploy trained models to physical robot systems.
  • Perform real-world robot debugging, testing, and performance optimization.
  • Diagnose issues across models, sensors, software, and robotic hardware.
  • Iterate between model training and real-world testing to improve system performance.
  • Help ensure models operate reliably and consistently in real-world environments.

Qualifications

Required

  • 1+ years of relevant industry or research experience in machine learning, robotics, computer vision, embodied AI, or a related field.
  • Strong understanding of deep learning and modern machine learning methods.
  • Experience with PyTorch or similar deep learning frameworks.
  • Experience training and evaluating machine learning models.
  • Strong programming skills in Python and familiarity with relevant ML/robotics tooling.
  • Understanding of multimodal learning, computer vision, robotics, or related areas.
  • Ability to work in a fast-paced startup environment and take ownership of technical problems from research through implementation.
  • Strong problem-solving and debugging skills.

Strong Plus

Real-World Robotics Deployment

  • Experience deploying and debugging machine learning models on physical robot systems.
  • Ability to bring models from development into real-world robotic environments.
  • Experience troubleshooting and stabilizing robotic systems in production or experimental environments.

Multimodal / VLA Models

  • Experience working with force, tactile, or other multimodal sensing.
  • Experience designing or training multimodal fusion models.
  • Hands-on experience with VLA models, including model design, training, evaluation, or real-world applications.
  • Experience with robotic manipulation or embodied AI systems.

Large-Scale Distributed Training

  • Experience with large-scale distributed model training.
  • Familiarity with DDP, DeepSpeed, FSDP, or similar distributed training frameworks.
  • Experience optimizing training performance, GPU utilization, memory usage, or training throughput.
  • Experience working with large-scale datasets and distributed data pipelines.

Ideal Candidate

We are looking for an engineer who is excited about the intersection of foundation models and physical robotics.

The ideal candidate is:

  • Hands-on and comfortable moving between research, coding, experimentation, and real-world robot testing.
  • Interested in solving problems that cannot be addressed through simulation or software alone.
  • Comfortable working with large-scale datasets and modern foundation-model architectures.
  • Able to take ownership of a problem from data training evaluation deployment real-world iteration.
  • Comfortable working in an early-stage environment where priorities can move quickly.
  • Curious about emerging VLA, World Model, and embodied AI research and able to translate new ideas into working systems.

Why Join MaxInsights?

  • Work directly on embodied AI and robotics foundation models.
  • Work with large-scale real-world robotics and human demonstration data.
  • Gain hands-on experience across the full AI development lifecycle, from data pipelines to real-robot deployment.
  • Work in a fast-moving startup environment with significant ownership and technical autonomy.
  • Collaborate with teams working at the forefront of robotics and foundation-model development.

Default Benefits:

  • Health insurance
  • Vision care
  • Dental coverage
  • 401(k)
  • Paid holidays
  • PTO (Paid Time Off)
  • Sick leave

Role Details

Company MaxInsights
Title Embodied AI / VLA Research Engineer
Location Santa Clara, CA, US
Experience Mid Level
Salary Not disclosed
Remote No

About This Role

Research Engineers bridge the gap between research and production. They implement papers, build experiment infrastructure, optimize training pipelines, and make research prototypes production-ready. They're the engineers who make research work at scale.

The role sits at a unique intersection. You need to understand the math well enough to implement novel architectures correctly, and you need the engineering chops to make them run efficiently on distributed systems. When a research scientist has a breakthrough idea, you're the person who turns it from a notebook prototype into a training pipeline that runs on 256 GPUs.

Across the 9,079 AI roles we're tracking, Research Engineer positions make up 2% of the market. At MaxInsights, this role fits into their broader AI and engineering organization.

Research Engineer roles are growing as AI labs recognize that research velocity depends on engineering quality. The role is less competitive than Research Scientist (no PhD required), but the bar for engineering skill is very high. These roles are concentrated at major labs and well-funded startups.

What the Work Looks Like

A typical week involves: implementing a new attention mechanism from a recent paper, profiling and optimizing a training pipeline that's bottlenecked on data loading, building evaluation infrastructure for a new benchmark, debugging distributed training issues across a GPU cluster, and pair-programming with a research scientist on their latest experiment. The work is deeply technical.

Research Engineer roles are growing as AI labs recognize that research velocity depends on engineering quality. The role is less competitive than Research Scientist (no PhD required), but the bar for engineering skill is very high. These roles are concentrated at major labs and well-funded startups.

Skills Required

Python (18% of roles) Pytorch (5% of roles)

Strong software engineering fundamentals plus ML knowledge. Python, C++, and CUDA experience are common requirements. You'll need to read papers and turn ideas into working code. Distributed systems experience (especially distributed training) is highly valued. Performance optimization skills separate great candidates from good ones.

Experience with large-scale training infrastructure (FSDP, DeepSpeed, Megatron), GPU programming (CUDA, Triton), and the internals of ML frameworks (PyTorch internals, custom autograd functions) is what makes candidates stand out. The best research engineers can debug issues that span the full stack from GPU memory management to numerical precision to algorithmic correctness.

Strong postings mention the team's recent research, the infrastructure scale, and the specific technical challenges. They often list the research areas you'd support. Look for roles that emphasize both implementation quality and research understanding.

Compensation Benchmarks

Research Engineer roles pay a median of $252,000 based on 457 positions with disclosed compensation. Mid-level AI roles across all categories have a median of $193,200.

Across all AI roles, the market median is $213,000. Top-quartile compensation starts at $263,911. The 90th percentile reaches $318,400. For comparison, the highest-paying categories include AI Safety ($256,000) and AI Engineering Manager ($251,622). By seniority level: Entry: $111,000; Mid: $193,200; Senior: $229,300; Director: $272,300; VP: $235,830.

MaxInsights AI Hiring

MaxInsights has 4 open AI roles right now. They're hiring across Research Engineer. Based in Santa Clara, CA, US.

Location Context

Across all AI roles, 6% (531 positions) offer remote work, while 8,478 require on-site attendance. Top AI hiring metros: New York (3,078 roles, $221,000 median); San Francisco (2,646 roles, $261,000 median); Los Angeles (1,486 roles, $210,000 median).

Career Path

Common paths into Research Engineer roles include Software Engineer, ML Engineer, Research Intern.

From here, career progression typically leads toward Senior Research Engineer, Research Scientist, ML Architect.

This is one of the best entry points into AI research without a PhD. Build a strong engineering portfolio with ML projects, contribute to open-source ML frameworks, and demonstrate that you can implement complex ideas correctly and efficiently. The transition to Research Scientist is possible with published first-author work, which some research engineer roles support.

What to Expect in Interviews

Technical screens test both engineering skill and research understanding. Expect coding rounds with performance-critical implementations (GPU optimization, efficient data loading). Be prepared to discuss papers relevant to the team's research area and explain how you'd implement key ideas. System design questions focus on training infrastructure: distributed training, experiment tracking, and compute resource management.

When evaluating opportunities: Strong postings mention the team's recent research, the infrastructure scale, and the specific technical challenges. They often list the research areas you'd support. Look for roles that emphasize both implementation quality and research understanding.

AI Hiring Overview

The AI job market has 9,079 open positions tracked in our dataset. By seniority: 106 entry-level, 4,384 mid-level, 3,343 senior, and 1,246 leadership roles (Director, VP, C-Level). Remote roles make up 6% of the market (531 positions). The remaining 8,478 roles require on-site or hybrid attendance.

The market median for AI roles is $213,000. Top-quartile compensation starts at $263,911. The 90th percentile reaches $318,400. Highest-paying categories: AI Safety ($256,000 median, 66 roles); Research Engineer ($252,000 median, 457 roles); AI Engineering Manager ($251,622 median, 55 roles).

Research Engineer roles are growing as AI labs recognize that research velocity depends on engineering quality. The role is less competitive than Research Scientist (no PhD required), but the bar for engineering skill is very high. These roles are concentrated at major labs and well-funded startups.

The AI Job Market Today

The AI job market spans 9,079 open positions across 16 role categories. The largest categories by volume: AI/ML Engineer (6,862), AI Product Manager (453), AI Software Engineer (451). These three account for the majority of open positions, though smaller categories often have higher per-role compensation because of specialized skill requirements.

The seniority mix tells a story about where AI teams are in their maturity. Entry-level roles (106) are outnumbered by mid-level (4,384) and senior (3,343) positions, reflecting that most companies are past the 'build a team from scratch' phase and need experienced engineers who can ship production systems. Leadership roles (Director, VP, C-Level) total 1,246 positions, representing the bottleneck between technical execution and organizational strategy.

Remote work availability sits at 6% of all AI roles (531 positions), with 8,478 requiring on-site or hybrid attendance. The remote share has stabilized after the post-pandemic correction. Senior and specialized roles (Research Scientist, ML Architect) are more likely to be remote-eligible than entry-level positions, partly because experienced hires have more negotiating power and partly because these roles require less hands-on mentorship.

AI compensation is structured in clear tiers. The market median sits at $213,000. Top-quartile roles start at $263,911, and the 90th percentile reaches $318,400. These figures include base salary with disclosed compensation. Total compensation (including equity, bonuses, and sign-on) runs 20-40% higher at companies that offer those components.

Category matters for compensation. AI Safety roles lead at $256,000 median, while Prompt Engineer roles sit at $141,440. The spread between highest and lowest-paying categories reflects the premium on specialized technical skills versus broader analytical roles.

The most in-demand skills across all AI postings: Python (1,649 postings), Aws (898 postings), Azure (723 postings), Rag (671 postings), Gcp (536 postings), Pytorch (434 postings), Claude (410 postings), Kubernetes (406 postings). Python dominates, appearing in the vast majority of role descriptions regardless of category. Cloud platform experience (AWS, GCP, Azure) is the second most common requirement. The newer entrants to the top skills list (RAG, vector databases, LLM APIs) reflect the shift from traditional ML toward generative AI applications.

Frequently Asked Questions

Based on 457 roles with disclosed compensation, the median salary for Research Engineer positions is $252,000. Actual compensation varies by seniority, location, and company stage.
Strong software engineering fundamentals plus ML knowledge. Python, C++, and CUDA experience are common requirements. You'll need to read papers and turn ideas into working code. Distributed systems experience (especially distributed training) is highly valued. Performance optimization skills separate great candidates from good ones.
About 6% of the 9,079 AI roles we track offer remote work. Remote availability varies by company and seniority level, with senior and leadership roles more likely to offer location flexibility.
MaxInsights is among the companies actively hiring for AI and ML talent. Check our company profiles for detailed breakdowns of open roles, salary ranges, and hiring trends.
Common next steps from Research Engineer positions include Senior Research Engineer, Research Scientist, ML Architect. Progression depends on whether you lean toward technical depth, people management, or product strategy.

Get Weekly AI Career Intelligence

Salary data, skills demand, and market signals from 16,000+ AI job postings. Every Monday.