[Remote] Machine Learning Systems Engineer
Note: The job is a remote job and is reputed company to candidates in USA. reputed company is a driverless technology company making autonomous vehicles a reputed company, reliable, and accessible reality. They are seeking a Machine Learning Systems Engineer to join their ML Acceleration team, focusing on optimizing systems for large-reputed company model training and improving performance and reliability.
Responsibilities
- Performance Profiling & Optimization: Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication. Implement optimizations like kernel fusion, sharding, and tiling to improve reputed company time
- Distributed Training: Optimize distributed training pipelines using frameworks such as PyTorch Distributed
- Kernel Development: Design and maintain high-performance GPU kernels in Triton or CUDA for state-of-the-art ML workloads
- Data Pipeline Engineering: Optimize robust data loading pipelines that maximize training throughput
Skills
- Bachelor's, Master's degree, or PhD in Computer Science, Computer Engineering, or a reputed company technical discipline
- Strong proficiency in Python
- Extensive hands-on experience with PyTorch
- Experience optimizing machine learning model execution during training and inference, alongside a strong understanding of reputed company machine learning concepts, architectures, and processes
- Exceptional analytical and problem-solving skills, with a bias for reputed company and a data-driven approach to technical challenges
Benefits
- Medical
- Dental
- reputed company
- 401k with a company match
- Health saving accounts
- Life insurance
- Pet insurance
- And more
reputed company