RemoteLeaf Logo
Log in Sign up

Machine Learning Systems Engineer

2 months, 1 week ago
United States
Full-time
Mid Level
Machine Learning Engineer
DevOps and Infrastructure
Machine Learning Python PyTorch
Apply Now
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
View All Jobs 44

Description

  • Profile and optimize training performance by identifying bottlenecks in data loading, gradient computation, and communication.
  • Implement system-level 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 for machine learning workloads.
  • Engineer robust data loading pipelines that maximize training throughput.
  • Work at the intersection of machine learning research and high-performance systems engineering to support large-scale distributed training.
  • Improve core infrastructure that enables researchers to train frontier models at scale.

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

Benefits

  • Hybrid schedule with in-office time in Boston, Pittsburgh, or Las Vegas, or the option to work fully remote.
  • Base salary range of $144,000 to $192,000 USD.
  • Additional compensation may include a bonus or company equity.
  • Medical, dental, and vision insurance.
  • 401(k) with company match.
  • Health savings accounts.
  • Life insurance and pet insurance.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

x.ai

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.

United States Full-time Junior Infrastructure Engineer Machine Learning Engineer
$180k-$440k
Ansible C++ Linux Puppet Python PyTorch Rust
15 hours, 33 minutes ago
Reddit

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.

United States Full-time Senior Machine Learning Engineer
$217k-$303k
Deep Learning LLM Machine Learning NLP Python PyTorch TensorFlow
16 hours, 3 minutes ago
Torc

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.

United States Full-time Senior Machine Learning Engineer
$177k-$213k
C++ Computer Vision Deep Learning Machine Learning Python PyTorch
16 hours, 18 minutes ago
CloudLinux

Security Engineer - Node.js Proactive Defense (worldwide remote, work anywhere)

CloudLinux 51-250 IT Services

CloudLinux is hiring an engineer-researcher to build a new runtime protection product for Node.js web apps running on managed hosting, from design through production rollout.

Poland Portugal Romania Bulgaria Spain Czech Republic Full-time Lead Research Scientist Security Engineer
JavaScript Node.js PHP WAF WordPress
16 hours, 33 minutes 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