ML Systems Engineer, ML Acceleration

2 months, 3 weeks ago
Full-time
Mid Level
Software Development
Motional

Motional

Motional is a leading company in driverless technology and autonomous vehicles, leveraging decades of industry expertise to develop and deploy safe and reliable autonomous vehicles. With a powerful DNA combining Aptiv's automotive technology and Hyunda...

Automotive
1K-5K
Founded 2020
$20M raised

Description

  • Profile and optimize training bottlenecks in data loading, gradient computation, and communication using tools such as Nsight and PyTorch Profiler.
  • Implement performance improvements including kernel fusion, sharding, and tiling to reduce step time.
  • Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
  • Design, develop, and maintain high-performance GPU kernels in Triton or CUDA for machine learning workloads.
  • Improve and maintain data loading pipelines to maximize training throughput.
  • Work at the intersection of machine learning research and high-performance systems engineering to support large-scale distributed training.
  • Contribute to reducing time-to-convergence for next-generation model training.
  • Focus on improving speed, cost, reliability, and throughput of core ML systems.

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.
  • Bias for action and a data-driven approach to technical challenges.
  • Experience with profiling tools such as Nsight or PyTorch Profiler (preferred).
  • Experience with Triton, CUDA, or distributed training frameworks such as PyTorch Distributed (preferred).

Benefits

  • This posting does not list any specific benefits or perks.
  • Motional emphasizes a progressive, global, diverse, and inclusive workplace.
  • Headquartered in Boston with operations in the U.S. and Asia.
  • Participation in E-Verify for newly hired employees.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

ML Infrastructure Engineer

x.ai 51-250 Internet Software & Services

SpaceXAI is hiring an ML Infrastructure Engineer to build and optimize the machine learning platform that powers recommendations on X.

Ansible C++ Linux Puppet Python PyTorch Rust
15 hours, 34 minutes ago

Senior Machine Learning Engineer, Safety

Reddit 1K-5K Internet Software & Services

Reddit is hiring a remote-friendly Machine Learning Engineer to build and improve safety systems that support enforcement of Reddit rules using large language models.

Deep Learning LLM Machine Learning NLP Python PyTorch TensorFlow
16 hours, 4 minutes ago

Senior, Machine Learning Engineer - 3D Perception

Torc 251-1K Road & Rail

Torc is hiring a Senior Machine Learning Engineer – 3D Perception to develop and deploy production perception models for autonomous trucks and improve Bird's Eye View understanding across its autonomy stack.

C++ Computer Vision Deep Learning Machine Learning Python PyTorch
16 hours, 19 minutes ago

Applied AI Engineer

Unframe Inc. 51-200 Technology, Information and Internet

Unframe is hiring an Applied AI Engineer to work with enterprise customers and internal teams on deploying AI-driven solutions that connect business problems to production systems.

Machine Learning Python React TypeScript
1 day, 14 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers