Machine Learning Applications and Compiler Engineer, New College Grad

Machine Learning EngineerMachine Learning EngineerFull TimeRemoteSeniorTeam 10,001+Since 1993H1B SponsorCompany SiteLinkedIn

Location

Canada

Posted

38 days ago

Salary

$135K - $220K / year

Seniority

Senior

Postgraduate DegreeEnglishPyTorchRustTensorflow

Job Description

Machine Learning Applications and Compiler Engineer, New College Grad

NVIDIA

• Build, develop, and maintain high-performance runtime and compiler components, focusing on end-to-end inference optimization. • Define and implement mappings of large-scale inference workloads onto NVIDIA’s systems. • Extend and integrate with NVIDIA’s SW ecosystem, contributing to libraries, tooling, and interfaces that enable seamless deployment of models across platforms. • Benchmark, profile, and monitor key performance and efficiency metrics to ensure the compiler generates efficient mappings of neural network graphs to our inference hardware. • Collaborate closely with hardware architects and design teams to feedback software observations, influence future architectures, and codesign features that unlock new performance and efficiency points. • Prototype and evaluate new compilation and runtime techniques, including graph transformations, scheduling strategies, and memory/layout optimizations tailored to spatial processors. • Publish and present technical work on novel compilation approaches for inference and related spatial accelerators at top tier ML, compiler, and computer architecture venues.

Job Requirements

  • Pursuing or recently completed a MS or PhD in Computer Science, Electrical/Computer Engineering, or related field, or equivalent experience.
  • Possess software engineering background with familiarity in systems level programming (e.g., C/C++ and/or Rust) and solid CS fundamentals in data structures, algorithms, and concurrency.
  • Hands on experience with compiler or runtime development, including IR design, optimization passes, or code generation.
  • Experience with LLVM and/or MLIR, including building custom passes, dialects, or integrations.
  • Familiarity with deep learning frameworks such as TensorFlow and PyTorch, and experience working with portable graph formats such as ONNX.
  • Understanding of parallel and heterogeneous compute architectures, such as GPUs, spatial accelerators, or other domain specific processors.
  • Strong analytical and debugging skills, with experience using profiling, tracing, and benchmarking tools to drive performance improvements.
  • Excellent communication and collaboration skills, with the ability to work across hardware, systems, and software teams.

Benefits

  • Eligible for equity
  • Health insurance
  • Stock options
  • Paid time off
  • Professional development opportunities

Related Job Pages

More Machine Learning Engineer Jobs

Zendesk logo

Staff Machine Learning Engineer

Zendesk

The intelligent heart of customer experience.

Full TimeRemoteTeam 5,001-10,000Since 2007H1B Sponsor

• Own the models and algorithms at the heart of Routing & Presence • Plan and scope ML work with observability and iterability designed in from the start • Propose and evaluate alternatives • Design experimentation frameworks tailored to routing outcomes • Lead innovation sessions with Product ahead of the product development cycle • Identify new data sources needed for learning and evaluation • Combine multiple signals into dynamic routing decisions • Shape the boundary between classical ML and LLM-driven components • Anticipate issues across the full lifecycle • Work across multiple teams and systems • Mentor scientists and engineers • Look ahead 18+ months • Build reputation as a subject-matter expert

Ireland
Airbnb logo

Senior Staff Machine Learning Engineer, Post Training

Airbnb

Airbnb is a community based on connection and belonging.

Full TimeRemoteTeam 5,001-10,000Since 2007H1B Sponsor

• Work with large scale structured and unstructured data; explore, experiment, build and continuously improve foundation models for Airbnb product, business and operational use cases. • Create a multi-year tech roadmap that enables our team to stay on the leading edge of the rapidly evolving AI landscape and leverage the best in class technologies to deliver customer benefits. • Continuously evaluate recent and upcoming large foundational models, ensuring the selection and refinement of the highest quality models for enhanced performance and efficiency. • Hands-on prototype, develop and productionize LLM models and pipelines at scale, including both batch and real-time use cases. • Drive key AI architectural decisions for products, and contribute to Airbnb’s ML platform architecture and strategy.

United States
$248K - $310K / year
Revelation Pharma logo

Machine Learning Engineer

Revelation Pharma

Revelation Pharma | National Network of 503A & 503B Compounding Pharmacies 💊

Full TimeRemoteTeam 201-500Since 2020H1B No Sponsor

• Build the evaluation and validation framework for all agent-driven clinical recommendations • Develop patient risk stratification models for adherence prediction, adverse event likelihood, dosage titration optimization, and churn/dropout risk • Implement and manage the predictive analytics pipeline on AWS • Design and build the RAG architecture that grounds agent responses in clinical protocols • Own model lifecycle management • Build explainability layers for clinical recommendations • Collaborate with the clinical team to translate clinical protocols and pharmacy domain knowledge into model features, training labels, and validation criteria • Establish model monitoring and alerting

United States
$120K - $145K / year
Job Closed
Nestle logo

ML Ops Engineer

Nestle

Nestle is a globally recognized brand and the largest food company in the world. Offering a wide variety of food, nutrition, health, and wellness products, Nest

• Optimize, standardize and implement data science and machine learning solutions at scale and in cloud-based environments (Azure) • Participate in the end-to-end lifecycle of data science projects through the use of DevOps, code, experiment and model management, CI/CD and further industry best practices • Write well-designed, testable, efficient code • Work closely with data engineering to continuously improve the way we consume data and deploy models in production • Design and lead on monitoring, troubleshooting, debugging and incident management for our ML pipelines • Be a trusted advisor and evangelist to the team and stakeholders on various aspects of ML Ops, from scaling and throughput, to infrastructure and deployment strategies

Poland