Machine Learning Systems Engineer
New
M
MotionalAutonomous vehicles
This role can be fully remoteFull-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.
- Ability to work in a hybrid schedule or remotely.
Responsibilities
- Identify bottlenecks in data loading, gradient computation, and communication using profiling tools like Nsight and PyTorch Profiler.
- 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.
- Develop robust data loading pipelines to maximize training throughput.
View Full Description & ApplyYou'll be redirected to the employer's site
