NVIDIA
GPU-accelerated computing • artificial intelligence • deep learning • virtual reality • gaming
ML Compiler Engineer
December 5
NVIDIA
GPU-accelerated computing • artificial intelligence • deep learning • virtual reality • gaming
Description
• Develop groundbreaking technologies in machine learning compilers and AI systems • Innovate and develop new machine learning compiler and systems technologies • Design, implement, and optimize compilers for high impact AI workloads • Co-design learning system solutions with current and future ML compiler and algorithm technologies • Collaborate closely with other engineering teams at NVIDIA to build high impact solutions for machine learning acceleration
Requirements
• Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD are preferred • 7+ years of experience • Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX etc) • Strong python and C/C++ programming skills • Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX • Expertise in machine compilers (e.g. Apache TVM, MLIR) • Strong experience in GPU performance optimizations • Strong experience machine learning systems research and productionization • Open source project ownership or contributions
Benefits
• Equity and benefits • Diverse work environment • Equal opportunity employer
Apply NowSimilar Jobs
December 5
Seeking an experienced Engineer to build high-performance applications for GEICO. Focus on technical leadership and collaboration across teams.
December 4
As an Application Engineer at Waymark, lead ML/AI-driven workflows and collaborate on data solutions. Join our effort to enhance healthcare for Medicaid beneficiaries.
November 27
MBL Technologies Inc.
11 - 50
Join MBL Technologies as a CyberArk Engineer. Support IAM projects for government clients.
November 26
Join Thermo Fisher Scientific as a Field Service Engineer. Install and maintain LSMS and Chromatography systems for customer success.
November 25
Join Velera as a Certificate Services & Data Encryption Engineer to oversee cryptographic management and data security services. Ensure compliance and secure management of digital certificates.