Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Engineering Manager, ML Kernel Performance

Amazon Locker

Amazon Annapurna Labs in Cupertino seeks an Software Engineering Manager to lead ML kernel performance initiatives within AWS Neuron. You will mentor engineers and drive kernel optimizations, shaping compiler strategies and ML accelerator workloads.

Lead cross-functional efforts across frameworks, compilers, and runtime layers, partnering with customers to optimize their models on Inferentia and Trainium, while delivering scalable, high-impact solutions.

#J-18808-Ljbffr
Vacancy posted 12 hours ago
Similar jobs that could be interesting for youBased on the Engineering Manager, ML Kernel Performance in Cupertino, CA vacancy
  •  ...Annapurna Labs (U.S.) Inc. is seeking kernel engineers to optimize ML workloads on AWS accelerators. You will design high-performance compute kernels for Neuron, analyze multi-generation performance, and implement compiler optimizations in collaboration with compiler... 
    Performance

    Amazon Web Services (AWS)

    Cupertino, CA
    12 hours ago
  • $207k - $300k

     ...with them to discuss performance, development, provide...  ...code developed by other engineers and provide feedback to...  ...in a people management or team leadership role...  ...Familiarity with the Linux kernel interface, containers,...  ...a part of the Borglet ML, you will play a critical... 
    Performance
    Worldwide

    Google

    Sunnyvale, CA
    2 days ago
  • $168k - $270.25k

     ...modern AI. Today, NVIDIA high-performance processing platforms power...  ...reasoning, and large-scale systems engineering. To address these complex...  ...we are seeking an Engineering Manager to spearhead our strategy for...  ...development, CUDA software, ML framework, runtime, product,... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  • $224k - $356.5k

     ...standard for assessing LLM serving performance across various inference frameworks....  ...efficiency, and scaling. As Technical Lead Manager, you will lead the engineering team within NVIDIA’s Dynamo...  ...performance-critical infrastructure, ML tooling, or distributed systems.3+ years... 
    Performance
    Full time
    Local area
    Remote work
    Worldwide

    Nvidia

    Santa Clara, CA
    4 days ago
  • $207k - $301k

    People Management and Talent Development: Lead, mentor, and grow a high-performing team of systems and ML engineers. Drive a culture of excellence, psychological safety, and continuous learning. Guide career paths, define OKRs, and conduct performance evaluations.Strategic... 
    Performance

    Google

    Sunnyvale, CA
    2 days ago
  • $218k - $335k

    Job DescriptionRoleAs an Applied ML Validation Manager on the Software Validation team within the...  ..., continuous release, and long-term performance monitoring.About the Organization The...  ...Machine Learning, Robotics, Software Engineering, Data Science, or a related field.2+... 
    Performance
    Full time
    Local area
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    7 hours ago
  • $188.9k - $330.5k

     ...getting hands-on with ML hardware acceleration...  ...in-class ML inference performance at the lowestcost in cloud...  ...come from a software engineering, ML research, or...  ...startedtutorials to deep-dive kernel optimization guides....  ...work autonomously and manage multiple initiatives... 
    Performance
    Local area
    Immediate start
    Worldwide
    Flexible hours

    Amazon

    Cupertino, CA
    4 days ago
  • $272k - $431.25k

     ...infrastructure that stores, manages, and serves exabytes of data...  ...infrastructure, enabling researchers and engineers to reliably store massive...  ...— while building a high-performance engineering culture grounded...  ...or prefetching tooling for AI/ML workloads — with direct... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    3 days ago
  • $141.8k - $258.6k

     ...Learning (GGML) team is seeking a Technical Program Manager to help shape the future of scalable machine...  ...execution, and cross‑functional coordination across engineering teams to deliver robust, high‑performance ML systems that operate across Apple’s cloud and distributed... 
    Performance
    Relocation

    Apple Inc.

    Cupertino, CA
    4 days ago
  • $147k - $299.5k

     ...the Netskope One platform, its Zero Trust Engine, and the powerful NewEdge network to gain full visibility and control without performance trade-offs.At Netskope, our technology is...  ...artificial intelligence (AI) and machine learning (ML) technology to power the Netskope One... 
    Performance

    Netskope

    Santa Clara, CA
    2 days ago
  • $207k - $300k

     ...challenges into clear algorithmic, ML, GenAI formulations,...  ...directly with product, UX, and engineering to define the optimal visual...  ...with individuals to discuss performance, development, feedback and coaching...  ...of experience in a people management, supervision/team leadership... 
    Performance

    Google

    Mountain View, CA
    2 days ago
  • $224k - $356.5k

    NVIDIA is seeking an Engineering Manager to lead the development of an agentic platform for observing...  ...into model behavior, inference performance, reliability, and cost across large-scale...  ...distributed systems, observability platforms, ML infrastructure, or production AI... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    2 days ago
  •  ...promote and monetize their work. Our platforms are highly performant, deployed at scale, and set new standards for enabling...  .... The Apple Ads EPM team is looking for a senior ML-focused Engineering Program Manager to lead complex initiatives across our machine learning... 
    Performance

    Socket.dev

    Cupertino, CA
    4 days ago
  • $272k - $431.25k

     ...computing platform relies on continuous performance excellence at every stage of...  ...an outstanding Performance Analysis Manager to lead an engineering team responsible for supervising and...  ...support. Your role involves analyzing kernel authoring flows from DSLs to internal... 
    Performance
    Full time
    Shift work

    Nvidia

    Santa Clara, CA
    1 day ago
  • $206.2k - $356.4k

     ...Machine Learning Engineering Manager - Ads Recommendations Cupertino, California, United States...  ...Store and Apple News, delivering highly-performant, privacy-first solutions that set new...  ...~ BS, or equivalent experience, in AI, ML, Mathematics, Computer Science, or a related... 
    Performance
    Worldwide
    Relocation

    Apple Inc.

    Cupertino, CA
    12 hours ago
  • $157.2k - $254.1k

     ...model security, posture management, AI red teaming, and...  ...Machine Learning Inference Engineer, you will serve as a...  ...of our AI platform - ML inference. Beyond...  ...set the standard for performance, scalability, and engineering...  ...like custom kernels, hardware acceleration... 
    Performance
    Full time
    Work at office

    Palo Alto Networks

    Santa Clara, CA
    4 days ago
  •  ...GenAI Inference Optimization Engineer to join our Models and Applications...  ...role focuses on improving performance, efficiency, and scalability...  ...across multiple layers—from kernels and runtimes to frameworks...  ...architectures.- Experience with ML frameworks (PyTorch, JAX, or... 
    Performance

    AMD

    San Jose, CA
    1 day ago
  •  ...a Principal Machine Learning Engineer to join our Models and Applications...  ...end-to-end training pipeline performance.Optimize the distributed...  ...EXPERIENCE:Experience with ML/DL frameworks such as PyTorch...  ...is a plus.Experience with GPU kernel optimization is a plus.Excellent... 
    Performance

    AMD

    San Jose, CA
    4 days ago
  • $272k - $431.25k

     ...NVIDIA is looking for a Machine Learning (ML) Engineer to join the GPU accelerated Apache Spark team. Apache Spark is the most popular...  ...be doing:Design and implement machine learning solutions for performance prediction and optimization of GPU accelerated enterprise Apache... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  • $207k - $340k

     ...LinkedIn, hybrid roles are performed both from home and from...  ...Staff Software Engineer to lead LinkedIn’s GPU-...  ...efficiency, batching, and kernel-level optimization. You...  ...precision, batching, memory management, and throughput or...  ....Suggested Skills:AI / ML Infrastructure Technical... 
    Performance
    For contractors
    Work at office
    Remote work
    Work from home
    Flexible hours

    Linkedin

    Mountain View, CA
    3 days ago
  • $296.3k

     ...minimum.The Role:We are seeking a Principal AI Engineer to lead the design and advancement of our...  ...:Architect, build, and optimize core AI/ML platform infrastructure to support...  ...practices for reliability, scalability, and performance across the AI/ML platform.Provide... 
    Performance
    Full time
    Local area
    Remote work
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  • $204k - $343k

     ...flexibility and trust our employees to manage their schedules responsibly....  .... About the role As an Engineering Manager on the Data...  ...annotation costs, and accelerate ML iteration cycles Set team goals...  ..., and growth for a high‑performing, mission‑driven team We’re looking... 
    Performance
    Full time
    For contractors
    For subcontractor
    Casual work
    Work at office
    Remote work
    Day shift

    Applied Intuition Inc.

    Sunnyvale, CA
    5 days ago
  •  ...Collaboration - Winning Spirit - High Performance\n\nAt Boston Scientific, we\u...  ...solutions while fostering engineering excellence and cross-...  ...n * Collaborate with product management, UX, and quality teams to deliver...  ...practices\n * Exposure to AI/ML technologies and data-driven... 
    Performance
    Hourly pay
    Full time
    Work at office
    Worldwide
    Relocation package
    Shift work
    3 days per week

    Boston Scientific

    Santa Clara, CA
    7 hours ago
  • $224k - $356.5k

     ...intelligence and autonomous vehicles. NvStreams is a high-performance inter-process and inter-chip communication library within NVIDIA...  ...the entire processing pipeline. We are seeking a seasoned Engineering Manager to lead our rapidly growing NvStreams team. If you are... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    4 days ago
  • $192k - $278k

     ...interfaces, compute, power, and performance.Execute custom silicon NPI,...  ...'s degree in Electrical Engineering, Computer Engineering, Computer...  ...speed interfaces.Experience as a manager or technical leadership for...  ...work to shape the future of AI/ML hardware acceleration. You... 
    Performance
    Contract work
    Worldwide

    Google

    Sunnyvale, CA
    2 days ago
  • $143k - $286k

     ...Summary...What you'll do...Principal Data Engineer - Agentic Data PlatformsThe MissionAs a...  ...Platform StrategyPartner with Engineering, AI/ML, Product, and Platform leaders to align...  ...at scale.Strong systems thinking and performance optimization skills.Influence Without AuthorityDemonstrated... 
    Performance
    Full time
    Temporary work
    Part time

    Sam's Club

    Sunnyvale, CA
    7 hours ago
  • $220k - $250k

     ...backgrounds and industries.Director of Engineering, Core DatabaseYugabyte is...  ...maintains industry-leading performance, scalability, and resilience. They will also manage and mentor our growing team.This...  ...integrating research concepts (e.g., ML-based anomaly detection) into... 
    Performance
    H1b
    Work at office
    Local area
    Worldwide
    Visa sponsorship
    2 days per week
    3 days per week

    Yugabyte

    Sunnyvale, CA
    7 hours ago
  • $220.92k - $311.89k

     ...experienced a Director, SoC Design Engineering, to lead the functional...  ...opportunity to shape the quality and performance of Intel products, ensuring...  ...;-5+ years of people management experience- Expertise in SystemVerilog...  ....- Experience with AI/ML accelerators, custom interconnect... 
    Performance
    Full time
    Work experience placement
    Local area
    Immediate start
    Worldwide
    Shift work
    Night shift

    Intel

    Santa Clara, CA
    2 days ago
  • $215k - $260k

     ...their AI strategies, and be part of a high-performing team that believes in each other, come...  ...About the Role: Join Crusoe as an Engineering Manager and lead a talented team focused on...  ...Intelligence Systems: Hands-on background in ML, heuristics, or rule-based systems —... 
    Performance
    Full time
    Temporary work

    Crusoe

    Sunnyvale, CA
    15 days ago
  • $168k - $258.75k

     ...optimize, and deploy GPU kernels automatically. This...  ...release faster, better-performing inference and training solutions.As Product Managers at NVIDIA, we enable...  ...partner closely with engineering, research, and customers...  ...platform products in AI, ML infrastructure, or... 
    Performance
    Full time

    Nvidia

    Santa Clara, CA
    3 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Engineering Manager, ML Kernel Performance. Be the first to apply!