about 5 hours ago
Las Vegas, NV, USAMid Level / Senior
H1B Sponsor
Base Salary
$144k - $192k/yr
Responsibilities
- Utilize profiling tools to identify bottlenecks in data loading and computation.
- Implement optimizations like kernel fusion and sharding to improve performance.
- Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
- Design and maintain high-performance GPU kernels for ML workloads.
- Optimize data loading pipelines to maximize training throughput.
Requirements
- Bachelor’s, Master’s degree, or PhD in Computer Science, Computer Engineering, or a related field.
- Strong proficiency in Python programming.
- Extensive hands-on experience with PyTorch.
- Experience optimizing machine learning model execution during training and inference.
- Exceptional analytical and problem-solving skills.
Benefits
- Medical, dental, and vision insurance.
- 401k with company match.
- Health savings accounts.
- Life and pet insurance.