Easy Apply
Easy Apply
The role involves building and maintaining AI infrastructure, optimizing training performance, and developing tools for robot training and data workflows.
Why RoboForce
RoboForce is an AI robotics company developing Physical AI–powered Robo-Labor for dull, dirty, and dangerous work. The company’s robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability.
We are looking for a Senior Software Engineer, AI Infra to build scalable AI infrastructure. As a Senior Software Engineer, you will architect and implement the core training infrastructure that enables large-scale model training, validation workflows, and production deployment for physical robots. You will work across cloud systems, GPU clusters, data pipelines, and robotics runtime environments to create a high-performance platform for Physical AI.
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.
Requirements
- 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.
Bonus Qualifications
- 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.
Top Skills
AWS
Azure
C++
Elasticsearch
GCP
Jax
Kubernetes
Python
PyTorch
Redis
Similar Jobs
Software
Design and build scalable backend systems for AI and GPU cluster observability, ensuring reliable performance and collaboration across teams.
Top Skills:
AWSAzureCC++DatabasesGCPGoJavaOpentelemetryPrometheusPython
Information Technology • Software
The Backend Engineer will contribute to building Ambient AI solutions, scale infrastructure, design AI pipelines, and collaborate closely with engineering and product teams to enhance model deployment and performance.
Top Skills:
DjangoDockerFastapiFlaskKubernetesPython
Artificial Intelligence • Computer Vision • Security • Software
Design and build machine learning infrastructure for real-time video processing and model training, collaborating with research scientists on AI advancements.
Top Skills:
AIComputer VisionDeep LearningDistributed SystemsMachine LearningPython
What you need to know about the Toronto Tech Scene
Although home to some of the biggest names in tech, including Google, Microsoft and Amazon, Toronto has established itself as one of the largest startup ecosystems in the world. And with over 2,000 startups — more than 30 percent of the country's total startups — Toronto continues to attract new businesses. Be it helping entrepreneurs manage their finances, simplifying business operations by automating payroll or assisting pharmaceutical companies in launching new drugs, the city's tech scene is just getting started.
