Senior Software Engineer, AI Infra

Sorry, this job was removed at 06:06 p.m. (CST) on Thursday, Sep 11, 2025
Easy Apply
Be an Early Applicant
Milpitas, CA
In-Office
Artificial Intelligence • Machine Learning • Robotics
The Role

We are looking for a Senior Software Engineer to build scalable AI infrastructure, supporting the training, validation, and production workflows for robotics. You will help establish robust systems for next-generation AI-driven robots.

Responsibilities:

  • Develop Python-based training infrastructure for stable and accelerated model training.
  • Build tools for data collection, training, and deployment of neural networks on RoboForce robots.
  • Optimize deployment pipelines for hardware platforms and maintain distributed systems.

Qualifications:

  • Bachelor’s or Master’s degree in Computer Science or related field with 8+ years of experience.
  • Proficiency in C++, Python, and ML frameworks (e.g., PyTorch, JAX).
  • Experience managing cloud infrastructure (AWS, Azure, GCP) and data stores (e.g., Postgres, MySQL, ElasticSearch, Redis).
  • Requires 5 days/week in-office collaboration with the teams.

Preferred Skills:

  • Expertise in profiling and optimizing CPU-GPU interactions.
  • Experience scaling neural network training jobs and GPU programming with CUDA.
  • Proven ability to develop annotation and dataset management tools.

Similar Jobs

Easy Apply
In-Office
Palo Alto, CA, USA
26 Employees

RoboForce Logo RoboForce

Senior Software Engineer

Artificial Intelligence • Machine Learning • Robotics
Easy Apply
In-Office
Milpitas, CA, USA
14 Employees

Commure Logo Commure

Senior Software Engineer

Information Technology • Software
In-Office
Mountain View, CA, USA
159 Employees
140K-230K Annually

Ambient.ai Logo Ambient.ai

Senior Software Engineer

Artificial Intelligence • Computer Vision • Security • Software
In-Office or Remote
7 Locations
105 Employees
168K-210K Annually
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
Milpitas, , California
14 Employees
Year Founded: 2023

What We Do

RoboForce, an AI-Robotics company, is building a first-of-its-kind Robotic Workforce System to take on the most tedious, force-demanding, and dangerous work humans don't have to do

Similar Companies Hiring

Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees
Fairly Even Thumbnail
Software • Sales • Robotics • Other • Hospitality • Hardware
New York, NY
Bellagent Thumbnail
Artificial Intelligence • Machine Learning • Business Intelligence • Generative AI
Chicago, IL
20 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account