Machine Learning Applications and Compiler Engineer, New College Grad
Location
Canada
Posted
38 days ago
Salary
$135K - $220K / year
Seniority
Senior
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 Guides
Related Job Pages
More Machine Learning Engineer Jobs
• 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
Senior Staff Machine Learning Engineer, Post Training
AirbnbAirbnb is a community based on connection and belonging.
• 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.
Machine Learning Engineer
Revelation PharmaRevelation Pharma | National Network of 503A & 503B Compounding Pharmacies 💊
• 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
ML Ops Engineer
NestleNestle 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




