Software Engineer, II - ML Frameworks
Torc Robotics
Software Engineering, Data Science
Ann Arbor, MI, USA
USD 139k-166,800 / year
Posted on Sep 23, 2025
About the Company
At Torc, we have always believed that autonomous vehicle technology will transform how we travel, move freight, and do business.
A leader in autonomous driving since 2007, Torc has spent over a decade commercializing our solutions with experienced partners. Now a part of the Daimler family, we are focused solely on developing software for automated trucks to transform how the world moves freight.
Join us and catapult your career with the company that helped pioneer autonomous technology, and the first AV software company with the vision to partner directly with a truck manufacturer.
Meet the Team
Torc's virtual driver software utilizes cutting-edge deep learning techniques to perceive the vehicle's environment, predict the movements of other vehicles, and execute accurate driving decisions. The ML Frameworks Team is seeking a motivated Software Engineer 2 to help build and maintain our machine learning training infrastructure. This is an excellent opportunity to contribute to the future of the autonomous vehicle industry while developing your skills in machine learning operations and distributed systems.
The ML Frameworks Team is hiring a Software Engineer 2 who will support the development and maintenance of our ML training framework components for distributed model training in the cloud. The engineer will assist in implementing features for our training architecture based on Ray and PyTorch Lightning, helping to migrate existing implementations towards this new architecture. This framework utilizes cloud resources for efficient model training and supports the training of multitask architectures for various perception and planning functions of the autonomous truck. Additionally, the engineer will help build tooling for various parts of the ML lifecycle and provide support to our internal users.
What You'll Do
-
Implement and maintain machine learning training pipelines and infrastructure
-
Support the development of our Ray-based framework for distributed machine learning training and data transformation pipelines
-
Help maintain the shared ML codebase used by all machine learning projects at Torc
-
Collaborate with researchers and engineers to support their machine learning workflows
-
Participate in building internal tools to enhance ML operations and developer productivity
-
Write clean, well-documented code and participate in code reviews
-
Learn and apply best practices for cloud-based ML infrastructure using AWS services
-
Participate in the on-call rotation to support the health of our services and framework
-
Work with senior engineers to troubleshoot and optimize training pipelines
What We're Looking For
-
Bachelor's degree in computer science, data science, artificial intelligence or related field with 1+ years of professional experience.
-
Good knowledge of Python and familiarity with PyTorch or other ML frameworks
-
Experience with AWS services (S3, Lambda, EC2, or similar cloud platforms)
-
Understanding of software engineering best practices including version control (Git) and agile workflows
-
Experience working with data processing pipelines or APIs
-
Strong problem-solving skills and ability to learn new technologies quickly
-
Excellent collaboration skills and ability to work in a team environment
-
Familiarity with AGILE development practices and tools such as Git and Jira
Bonus Points!
-
Previous experience with ML Ops workflows including dataset management, model training pipelines, and model registries
-
Exposure to distributed computing concepts or frameworks like Ray
-
Experience with PyTorch Lightning or similar training frameworks
-
Knowledge of containerization technologies (Docker, Kubernetes)
-
Experience in the autonomous vehicle space
-
Familiarity with GPU programming or optimization
This role offers an excellent opportunity to grow your expertise in ML infrastructure while working on cutting-edge autonomous vehicle technology. You'll be mentored by senior engineers and have the chance to make meaningful contributions to our ML training systems.
Job ID: 102383
Hiring Range for Job Opening
US Pay Range
$139,000—$166,800 USD