Job Description

Torc is seeking a Staff Machine Learning Engineer to enhance AI performance for autonomous trucks. The role involves developing a new distributed training architecture based on Ray and PyTorch Lightning, maintaining a large ML codebase, and collaborating with researchers and engineers.

Responsibilities

• Mature and optimize machine learning workflows
• Implement and roll out a new Ray-based framework for distributed, large-scale machine learning training
• Maintain a large codebase
• Collaborate with researchers and engineers
• Engage with data and compute interfaces
• Stay abreast of the latest advancements in PyTorch
• Develop innovative and performant deep learning solutions
• Analyze and optimize deep learning training
• Contribute to the development of internal tools and libraries
• Document work clearly and concisely

Requirements

• Bachelor's degree in a related field with 8+ years of experience or a master's degree with 5+ years of experience
• Mastery of Python and Pytorch
• Expert knowledge of Ray
• In-depth knowledge of AWS EC2 and Sagemaker
• Excellent understanding of parallel computing and high-performance concepts
• Excel at working in a highly collaborative environment
• Familiarity with AGILE development practices
• Comfortable using collaborative development tools such as Git and Jira
• Ability to adhere to company coding standards
• Proven dedication to writing production-quality code

Benefits

• No benefits