Machine Learning Systems Engineer
New
M
MotionalAutonomous Vehicles
Boston, Massachusetts, United States; Las Vegas, Nevada, United States; Pittsburgh, Pennsylvania, United States; Remote U.S.Full-Time
Salary$144,000 — $192,000 USD
Apply NowOpens the employer's application page
Job Details
- Required Skills
- PythonMachine LearningPyTorch
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.
- A bias for action and a data-driven approach to technical challenges.
Responsibilities
- Utilize profiling tools like Nsight and PyTorch Profiler to identify bottlenecks in data loading, gradient computation, and communication.
- Implement optimizations such as kernel fusion, sharding, and tiling to improve step time.
- Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
- Design and maintain high-performance GPU kernels in Triton or CUDA for state-of-the-art ML workloads.
- Optimize robust data loading pipelines that maximize training throughput.
View Full Description & ApplyYou'll be redirected to the employer's site
