← All roles

Research Engineer

Mind Robotics· Software Engineering
Apply on company site
Palo AltoFull-timePosted 6mo ago

About the role

Mind Robotics is building generalized physical AI for real-world industrial deployment, focusing on scalable ML infrastructure to enable fast, reliable model training and production deployment on the factory floor.

Responsibilities

  • Design and implement scalable systems for training large ML models.
  • Enable efficient workflows for data ingestion, training, and iteration.
  • Develop and optimize distributed training systems across hundreds of GPUs.
  • Implement strategies for parallelization, sharding, and efficient compute utilization.
  • Improve training efficiency through techniques such as attention optimizations, kernel fusion, and memory management.
  • Partner closely with modeling teams to accelerate iteration speed and reduce training costs.
  • Build internal tools for experiment tracking, monitoring, and debugging.
  • Implement systems for tracking training performance, failures, and resource utilization.
  • Debug and resolve bottlenecks across the training stack.
  • Provide lightweight infrastructure support for deploying and running models in production environments.
  • Optimize inference performance and reliability where needed.
  • Support core cloud infrastructure needs for training workloads (without heavy DevOps overhead).
  • Manage compute resources efficiently across training jobs.

Qualifications

  • Strong experience building infrastructure for large-scale ML training.
  • Deep understanding of how modern LLM/VLM systems are trained and scaled.
  • Proven experience setting up and scaling distributed training across hundreds of GPUs.
  • Strong understanding of parallelization strategies (data, model, pipeline parallelism).
  • Strong proficiency in Python programming.
  • Expert-level proficiency in PyTorch and/or JAX.
  • Strong understanding of techniques like attention optimization, kernel fusion, and efficient memory usage.

Skills

Required:PythonPyTorch and/or JAXDistributed trainingParallelization strategiesMemory optimization
Last updated Aug 19, 2026. Interested? Apply on the company site.

About Mind Robotics

Physical AI for the Real World

Mind Robotics is building intelligent robotics for industrial deployment. We believe the fastest path to broadly capable robots is through clearly defined, high-impact environments. We are starting where the need is most.

roboticsindustrial deploymentPhysical AImanufacturing
Company profileWebsite

More open roles at Mind Robotics