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 LearningPyTorchDistributed Systems
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.
- Data-driven approach to technical challenges.
Responsibilities
- Utilize profiling tools like Nsight and PyTorch Profiler to identify bottlenecks in data loading and gradient computation.
- Implement optimizations such as kernel fusion, sharding, and tiling to improve model training step times.
- Optimize distributed training pipelines using frameworks like PyTorch Distributed.
- Design and maintain high-performance GPU kernels in Triton or CUDA for ML workloads.
- Optimize robust data loading pipelines to maximize training throughput.
View Full Description & ApplyYou'll be redirected to the employer's site
