All roles

[Remote] Machine Learning Systems Engineer

Remote · USA Full-time New today

Note: The job is a remote job and is open to candidates in USA. reputed company is a driverless technology company making autonomous vehicles a safe, reliable, and accessible reality. They are seeking a Machine Learning Systems Engineer to join their ML Acceleration team, focusing on optimizing core systems for large-scale model training and enhancing performance across distributed training pipelines.

Responsibilities

  • 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
  • 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 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 action and a data-driven approach to technical challenges

Benefits

  • Candidates for certain positions are eligible to participate in reputed company’s benefits program.
  • reputed company’s benefits include but are not limited to medical, dental, reputed company, 401k with a company match, health saving accounts, life insurance, pet insurance, and more.

Company Overview

  • reputed company offers an autonomous driving platform for robotaxi providers, fleet operators, and automotive manufacturers. It is a sub-organization of Hyundai Motor Group. It was founded in 2020, and is headquartered in Boston, Massachusetts, USA, with a workforce of 501-1000 employees. Its website is https://reputed company.com.
  • Apply To This Job

    Related roles