Base Salary
$144k - $192k/yr
Responsibilities
- Use Nsight and PyTorch Profiler to identify bottlenecks in data loading, gradient computation, and communication.
- Implement performance optimizations such as kernel fusion, sharding, and tiling to improve training step time.
- Optimize distributed training pipelines using PyTorch Distributed.
- Design and maintain high-performance GPU kernels in Triton or CUDA.
- Optimize robust data loading pipelines to maximize training throughput.
Requirements
- Bachelor’s, Master’s degree, or PhD in Computer Science, Computer Engineering, or a related technical discipline.
- Strong proficiency in Python.
- Extensive hands-on experience with PyTorch.
- Experience optimizing machine learning model execution during training and inference.
- Strong understanding of fundamental machine learning concepts, architectures, and processes.
- Exceptional analytical and problem-solving skills with a data-driven approach to technical challenges.
Benefits
- Hybrid schedule with in-office time in Boston, Pittsburgh, or Las Vegas, or fully remote.
- Medical, dental, and vision insurance.
- 401k with company match.
- Health savings accounts.
- Life insurance.
- Pet insurance.
About Motional
Motional is an autonomous vehicle technology company on a mission to make driverless vehicles a safe, reliable, and accessible reality. The company formed in 2020 and is majority-owned by Hyundai Motor Group, one of the world’s largest vehicle manufacturers. Motional has extensive experience in commercial robotaxi operations, with service currently available on the Uber network in Las Vegas and plans for driverless operations by the end of 2026. Headquartered in Boston, Motional has operations in the U.S. and Singapore. To explore our current openings, please visit our careers page at motional.com/careers. Please be alert for job recruitment scams, including fraudulent job postings, offers, and communications by individuals or entities purporting to be affiliated with Motional. All of our third party job postings link back to our motional.com/careers page. Need assistance? If you have questions or want to report suspicious activity, send us a direct message - we’re here to help.