Mind Robotics Logo

Mind Robotics

Research Engineer

Reposted 6 Days Ago
In-Office
Palo Alto, CA, USA
Entry level
In-Office
Palo Alto, CA, USA
Entry level
The role involves building systems for large-scale model training, focusing on distributed training, ML infrastructure, and GPU performance optimization.
The summary above was generated by AI
About Mind:

Mind Robotics is building Physical AI for real-world industrial deployment, starting with the factory floor. We believe the hardest problems in AI are solved when researchers and engineers are hands-on with the physical world every day - and we're looking for people who are passionate about robotics, value ownership, and are excited to tackle difficult problems. Join us if you want to move beyond digital intelligence and put intelligence into motion.

About the team and the role:

At Mind Robotics, we’re building generalized physical AI—robotic systems capable of dexterous, adaptive, and reasoning-intensive work in real-world industrial environments. Our ability to iterate quickly on large-scale models depends on world-class ML infrastructure.

We’re looking for a Research Engineer to build the core systems that enable fast, reliable, and scalable model training—powering everything from experimentation to production deployment.

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.

Requirements:
  • 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.

Nice to Have:
  • Experience supporting inference systems in production.

  • Familiarity with robotics or embodied AI workloads.

  • Experience building tools for experiment management and researcher productivity.

Similar Jobs

6 Days Ago
Remote or Hybrid
USA
195K-290K Annually
Senior level
195K-290K Annually
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Lead design, build, and deploy of large-scale data platforms for LLMs, RAG, and agentic AI systems. Hands-on coding, architecting fault-tolerant pipelines, establishing MLOps/DataOps best practices, mentoring engineers, and operationalizing research into production across Exabyte-scale distributed systems.
Top Skills: AirflowAWSBigQueryDaskDevsecopsDockerFlinkGCPGoJvmKafkaKubeflowKubernetesLangchainLlamaindexLlmsMlflowOciPulsarPythonRetrieval-Augmented Generation (Rag)RustSagemakerSnowflakeSparkVertex Ai
16 Days Ago
Hybrid
128K-160K Annually
Mid level
128K-160K Annually
Mid level
Artificial Intelligence • Hardware • Software • Nanotechnology • Semiconductor • Quantum Computing • Defense
Lead research and development of agentic AI and multi-agent systems integrating LLMs, memory, planning, tool use, and GraphRAG-style retrieval. Develop graph machine learning and geometric deep learning pipelines, build knowledge-augmented AI (knowledge graphs, ontologies), and ensure trustworthy AI via XAI, V&V, robustness testing, and uncertainty quantification. Collaborate across teams, publish research, and support proposal development for mission-critical autonomous and decision-support applications.
Top Skills: Agent2Agent (A2A)AgentopsAutogenCypherDistributed InferenceGeometric Deep LearningGnnsGpu AccelerationGraphragKnowledge GraphsLanggraphLlmopsLpgModel Context Protocol (Mcp)Neo4JOntologiesPythonPyTorchRayRdfSglangSparkVllm
3 Days Ago
In-Office
Sunnyvale, CA, USA
207K-275K Annually
Senior level
207K-275K Annually
Senior level
Cloud • Information Technology • Machine Learning
Lead applied research to advance continuous learning for agents: design and evaluate LLM post‑training and RL methods, implement and deploy experiments at scale, validate research on customer tasks, optimize GPU/distributed training, and mentor engineers while driving cross‑functional technical direction.
Top Skills: CudaFastapiGpuKubernetesMegatronPostgresTemporal

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account