Senior Software Engineer, AI Infra

Posted 13 Hours Ago
Be an Early Applicant
Milpitas, CA
Senior level
Artificial Intelligence • Machine Learning • Robotics
The Role
In this role, the Senior Software Engineer will enhance the AI infrastructure for robotics by developing Python-based training and deployment tools, managing deep learning models, and ensuring robust backend systems. Close collaboration with AI Robotics experts is key to building both onboard and offboard solutions for next-gen robotics.
Summary Generated by Built In

Job overview:

We are seeking a highly skilled and experienced Senior Software to join our innovative AI robotics company, taking our AI infrastructure to the next level. In this role, you will build the robust offline workflows including training, validation, and production of the robot. You will work closely with AI Robotics expertise to build the robust onboard and offboard foundations for next-gen robot.

Why it’s interesting:

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.

At RoboForce, we're building a small team with a flat org structure and extreme talent density. Currently, we have top tech leaders from CMU Robotics, Michigan Robotics, Amazon Robotics, Tesla Robotics, Google, Waymo, Apple, and Microsoft. RoboForce is backed by world-class investors, including the Nobel Prize Laureate, Carnegie Mellon University, and beyond.

Responsibilities:

  • Design and develop Python-based training infrastructure to ensure stable and accelerated model training.
  • Design and implement software tools for data collection and management, training deep neural networks, and deploying them on RoboForce robots.
  • Develop and optimize tools for deploying trained neural networks to hardware platforms.
  • Design and maintain pipelines for running and validating PyTorch models.
  • Develop, operate, and maintain reliable and scalable backend distributed systems.

Minimum Qualifications:

  • Bachelor's or Master's degree in Computer Science, Robotics, Engineering, or a related field with at least 8 years of working experience.
  • Strong proficiency in both C++ and Python.
  • Descent experience with at least 2 low-level programming languages such as C, C++, or Rust.
  • Experience with Python and an ML framework, including PyTorch, JAX, TensorFlow, etc.
  • Experience using and managing data stores, including Postgres, MySQL, ElasticSearch, Redis, etc.
  • Experience in managing cloud infrastructure, including platforms such as AWS, Azure, and GCP.
  • Must work on-site at our office in the California Bay Area.
  • Must have and maintain US work authorization.

Preferred Qualifications:

  • Experience in profiling and optimizing CPU-GPU interactions, including techniques like pipelining compute and data transfers.
  • Experience in scaling neural network training jobs across GPU clusters.
  • Experience with GPU programming with CUDA
  • Experience in developing data annotation and dataset management tools.

Top Skills

C++
Python
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

Snap Inc. Logo Snap Inc.

Staff Software Engineer, ML Infrastructure, 9+ Years of Experience

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
4 Locations
5000 Employees
189K-333K Annually
Financial District, San Francisco, CA, USA
6282 Employees

Wayve Logo Wayve

Software Engineer, Machine Learning Platform

Artificial Intelligence • Transportation
Sunnyvale, CA, USA
200 Employees

NVIDIA Logo NVIDIA

Senior Software Engineer, Machine Learning VLSI Designs

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Santa Clara, CA, USA
21960 Employees

Similar Companies Hiring

RunPod Thumbnail
Software • Infrastructure as a Service (IaaS) • Cloud • Artificial Intelligence
Charlotte, North Carolina
53 Employees
HERE 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

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account