Machine Learning Systems Engineer
New
M
MotionalAutonomous vehicles
We encourage a hybrid schedule with in-office time at one of our locations in Boston, Pittsburgh, or Las Vegas to support collaboration, or this role can be fully remote.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.
- Data-driven approach to technical challenges.
Responsibilities
- Utilize profiling tools like Nsight and PyTorch Profiler to identify and resolve 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 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 to maximize training throughput.
View Full Description & ApplyYou'll be redirected to the employer's site
