Senior Software Engineer, AI Infra

Job Posted 17 Days Ago Posted 17 Days Ago
Be an Early Applicant
Milpitas, CA
Senior level
Artificial Intelligence • Machine Learning • Robotics
The Role
Develop AI infrastructure for training and deploying models. Optimize pipelines, manage data, and ensure stability in robotic applications.
Summary Generated by Built In

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).
  • Must be on-site with no remote work option.

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.

Top Skills

AWS
Azure
C++
Cuda
Elasticsearch
GCP
Jax
Ml Frameworks
MySQL
Postgres
Python
PyTorch
Redis
Am I A Good Fit?
beta
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
On-site Workplace
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 Jobs

Verkada Inc Logo Verkada Inc

Staff Backend Engineer - Core Services

Cloud • Hardware • Security • Software
San Mateo, CA, USA
2000 Employees
150K-280K Annually
Easy Apply
Hybrid
San Francisco, CA, USA
1100 Employees
174K-203K

Crunchyroll Logo Crunchyroll

Staff DevOps Engineer, Core Infrastructure Engineering

Digital Media • eCommerce • Gaming • Mobile • News + Entertainment
Hybrid
Los Angeles, CA, USA
1200 Employees
192K-240K Annually

Crunchyroll Logo Crunchyroll

Staff DevOps Engineer, Core Infrastructure Engineering

Digital Media • eCommerce • Gaming • Mobile • News + Entertainment
Hybrid
San Francisco, CA, USA
1200 Employees
213K-266K Annually

Similar Companies Hiring

HERE Technologies Thumbnail
Software • Logistics • Internet of Things • Information Technology • Computer Vision • Automotive • Artificial Intelligence
Amsterdam, NL
6000 Employees
True Anomaly Thumbnail
Software • Machine Learning • Hardware • Defense • Artificial Intelligence • Aerospace
Colorado Springs, CO
131 Employees
Caliola Engineering Thumbnail
Software • Machine Learning • Hardware • Defense • Data Privacy • App development • Aerospace
Colorado Springs, CO
53 Employees
By clicking Apply you agree to share your profile information with the hiring company.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account