ML Runtime Optimization Engineer

3 weeks, 6 days ago
Full-time
Mid Level
Software Development
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

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

Senior Software Engineer, Core Experiences - Fort Collins, CO, USA

Speechify 51-250 Internet Software & Services

Speechify is hiring a Senior Software Engineer for its Core Experiences Team to build the foundational services and SDKs that power its cross-platform product experience.

GCP Java Kotlin
10 minutes ago

Staff Software Engineer, Web

instacart.careers 1K-5K Internet Software & Services

Instacart is hiring a senior web platform leader to own the architecture behind its Home and Cross-Retailer Search experiences, powering customer discovery across the company’s grocery marketplace.

Design Systems Machine Learning Next.js React SEO TypeScript
14 minutes ago

Vice President, Software Engineer - Finance Engineering

Galaxy 251-1K Capital Markets

Galaxy is hiring a Python developer for its Financial Engineering team to build and support reporting solutions that inform finance and accounting decisions in a fast-paced digital assets environment.

Apache Spark AWS Blockchain C# Databricks Docker Go Java Kubernetes Pandas Python SQL
59 minutes ago

Software Engineer, Platform - Savannah, GA, USA

Speechify 51-250 Internet Software & Services

Speechify is hiring a Platform backend engineer to build and maintain core services for its distributed text-to-speech products, with a focus on scalable APIs, payments, subscriptions, and enterprise integrations.

AWS Azure Docker GCP Kubernetes Microservices Node.js REST API TypeScript
1 hour, 1 minute 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