RoboForce · 2 hours ago
Senior Software Engineer, AI Infra
Milpitas, CA
Full-time
Onsite
Senior Level
5+ years expRoboForce is an AI robotics company building Physical AI and Robo-Labor systems for dull, dirty, and dangerous work. They are seeking a Senior Software Engineer to build scalable AI infrastructure and architect core training infrastructure for large-scale model training and deployment for physical robots.
Artificial Intelligence (AI)Machine LearningRobotics
Hiring Manager
Stephany Vanhorn
Responsibilities
Build and maintain distributed training pipelines leveraging PyTorch, JAX, or equivalent frameworks across multi-GPU and multi-node clusters
Build tools for data collection, training, and deployment of neural networks on RoboForce robots
Architect robust cloud-native and on-prem GPU infrastructure across multi-cloud environments
Build high-throughput data workflows to support large-scale dataset ingestion, versioning, and distributed storage
Optimize end-to-end training performance: CPU–GPU transfers, NVMe caching, I/O pipelines, containerized runtime environments, and CUDA-level optimizations
Integrate training artifacts into on-robot inference stacks
Qualification
Required
Bachelor's or Master's degree in Computer Science or related field with 5+ years of experience
Strong proficiency with C++, Python, and ML frameworks (e.g., PyTorch, JAX)
Deep experience with at least one major cloud provider (GCP, AWS, Azure) and familiarity with Kubernetes, containerization, and GPU machine provisioning
Strong understanding of SQL and NoSQL data stores (Postgres, MySQL, BigQuery, ElasticSearch, Redis)
Requires 5 days/week in-office collaboration with the teams
Preferred
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
Benefits
Competitive stock options/equity programs.
Health, dental, and vision insurance, 401(k) plan.
Visa sponsorship and green card support for qualified candidates.
Lunches and dinners, a fully stocked kitchen, and regular team-building events.
Company
RoboForce
RoboForce is an AI Robotics startup building the world’s most advanced Robo-Labor system.
11-50 employees
H1B Sponsorship
RoboForce has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3)
2024 (4)
2023 (2)
Funding
Current Stage
Early StageTotal Funding
$15M2025-01-06Seed· $15M
Leadership Team
Leo Ma
Founder & CEO
Recent News
FTC Solar, Inc
2025-11-11
2025-10-16
Company data provided by crunchbase