RemoteLeaf Logo
Log in Sign up

ML Runtime Optimization Engineer

1 month, 2 weeks ago
United States
Full-time
Mid Level
Machine Learning Engineer
Software Development
Deep Learning Embedded Systems Machine Learning PyTorch
Apply Now
Applied Intuition

Applied Intuition

Vehicle software supplier | Applied Intuition Simulation and validation platform for automotive, trucking, construction, and more. Solutions to safely develop, test, and deploy autonomous systems at scale #PoweringAutonomy Industry leading platform tha...

Internet Software & Services
251-1K
Founded 2017
$352M raised
View All Jobs 36

Description

  • Drive ML performance optimization across multiple technologies for on-road and off-road ADAS/AD stacks on embedded compute platforms.
  • Develop compute usage strategies to improve inference efficiency and latency on customer-selected compute boards.
  • Work on model pruning and quantization to support deployment on memory-constrained platforms.
  • Collaborate closely with ML engineers and software developers to optimize model architecture solutions.
  • Set up profiling methodologies for target embedded platforms and identify performance bottlenecks during stack integration.
  • Support deployment of ML models across the full ML framework stack, including PyTorch, JAX, ONNX, TensorRT, CUDA, XLA, and Triton.

Requirements

  • Bachelor’s degree in Electrical Engineering, Computer Science, Mathematics, Physics, or a related field.
  • 3+ years of experience with ML accelerators, GPU, CPU, SoC architecture, and micro-architecture.
  • Strong software development skills with a focus on embedded programming.
  • Experience profiling and optimizing model performance on embedded compute platforms.
  • Experience working with deep learning frameworks such as PyTorch, JAX, and ONNX.
  • M.Sc. or PhD in a machine learning-related area is preferred.
  • Experience building an ML optimization framework from scratch is preferred.
  • Experience deploying ML solutions to embedded chips for real-time robotics applications is preferred.

Benefits

  • Base salary range of $159,053 - $199,295 USD annually.
  • Equity compensation in the form of options and/or restricted stock units.
  • Comprehensive health, dental, vision, life, and disability insurance coverage.
  • 401(k) retirement benefits with employer match.
  • Learning and wellness stipends.
  • Paid time off.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Cribl

Sr Machine Learning Engineer, AI Research

Cribl 251-1K IT Services

Cribl is hiring a machine learning engineer to help build AI-enabled security and observability products for enterprise customers as part of its telemetry platform.

United States Full-time Senior Machine Learning Engineer
$185k-$215k
Computer Vision Feature Engineering Kubeflow Machine Learning MLflow MLOps NLP Python PyTorch Reinforcement Learning TensorFlow
47 minutes ago
Odd.

AI/Machine Learning Engineer

Odd. 1-10 Consumer Goods

Oddball is hiring an AI/ML Engineer to build intelligent automation for a cloud-based federal case management system supporting a major financial regulatory agency.

United States Full-time Mid Level AI (Artificial Intelligence) Machine Learning Engineer
$120k-$160k
Agile AWS Java Machine Learning NLP Python PyTorch React Scikit-learn TensorFlow
1 hour, 6 minutes ago
SmithRx

Senior Software Engineer, Frontend - Platform

SmithRx 1K-5K Pharmaceuticals

SmithRx is hiring a Senior Front-End Software Engineer to help design, build, test, and maintain its web product suite for a fast-growing health-tech platform.

United States Full-time Senior Software Engineer
CI/CD CSS Cypress Docker Git GraphQL HTML JavaScript Microservices React Selenium UX Design
1 hour, 25 minutes ago
Unity

Senior Machine Learning Engineer, Ads Experimentation & Measurements

Unity 5K-10K Internet Software & Services

Unity is hiring a Senior Machine Learning Engineer to lead experimentation and evaluation for its Ads Experimentation Platform, improving how global ad models and delivery systems are validated and optimized at scale.

United States Full-time Senior Machine Learning Engineer
$149k-$259k
Apache Spark GCP Machine Learning Python Scala Snowflake Statistics
1 hour, 27 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