Logo
RoboForce

Senior Software Engineer, AI Infra

RoboForce, Milpitas

Save Job

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.

Apply for this job

*

indicates a required field

First Name *

Last Name *

Email *

Phone

Resume/CV *

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

#J-18808-Ljbffr