Senior Machine Learning Engineer for GPU Optimization

5 days ago


San Jose, California, United States Adobe Full time
Join Adobe's Team of Innovators

We're on a mission to revolutionize digital experiences and empower creators to bring their visions to life. As a Senior Machine Learning Engineer for GPU Optimization, you'll play a critical role in shaping the future of AI and machine learning at Adobe.

About the Role

We're seeking a highly skilled and experienced engineer to optimize our machine learning training processes for large-scale GPU deployments. Your expertise in CUDA, CuTe, and PyTorch will enable us to accelerate our AI research and development, driving innovation and excellence in our products.

Key Responsibilities
  • Design and implement efficient forward and backward passes in CUDA/CuTe
  • Develop optimized custom layers in PyTorch
  • Optimize ML training code for large, distributed training with FP8
  • Analyze and optimize performance between data types such as BF16 and FP8 for large deep learning models
  • Understand and optimize H100 GPUs
  • Architect broader, end-to-end optimized training code and schemes with PyTorch for large distributed models
Requirements
  • Proficiency in at least two of: Linux, Ansible, Docker, Kubernetes (5+ yrs)
  • Expert in Python and/or C++
  • Expert in CUDA/CuTe, OpenCL, and/or Triton
  • Expert in PyTorch
  • Experience with DDP, FSDP
  • Experience in distributed computing (7+ yrs)
  • Experience working with AWS or similar cloud infrastructure (5+ yrs)
  • Experience with HW resource management for ML training and/or deployment
  • Bachelor's or Master's degree in Computer Science, Computer Engineering, or a related area
What We Offer

At Adobe, we're committed to creating exceptional employee experiences and fostering a culture of innovation and collaboration. Our compensation reflects the cost of labor across several geographic markets, and we pay differently based on those defined markets. The pay range for this position is $170,900 -- $325,200 annually. Pay within this range varies by work location and may also depend on job-related knowledge, skills, and experience.

In addition to a competitive salary, you'll have access to a comprehensive benefits package, including medical, dental, and vision insurance, 401(k) matching, and paid time off. We're also committed to providing opportunities for growth and development, including training and education programs, mentorship, and career advancement opportunities.



  • San Jose, California, United States Adobe Full time

    Unlock the Power of Machine Learning with AdobeAt Adobe, we're on a mission to revolutionize the world of digital experiences. As a Sr. ML Training GPU Optimization Engineer, you'll play a critical role in empowering our teams to design and deliver exceptional digital experiences.What You'll Be Working On:Design and implement efficient forward and backward...


  • San Jose, California, United States QuantumScape Full time

    Job SummaryWe are seeking a highly skilled Senior Machine Learning Engineer to lead our team in developing advanced deep learning solutions for analyzing image and characterization data.This is a unique opportunity to join our team at QuantumScape and contribute to the development of cutting-edge technologies in the field of materials research and...


  • San Jose, California, United States Samsung Electronics Full time

    Job Title: Senior GPU Performance EngineerAt Samsung Electronics, we are seeking a highly skilled Senior GPU Performance Engineer to join our team. As a key member of our Xclipse GPU software team, you will be responsible for delivering cutting-edge technologies to revolutionize the mobile GPU market.Key Responsibilities:Optimize and fine-tune the...


  • San Francisco, California, United States San Francisco Compute Group Full time

    About San Francisco Compute GroupWe are a leading provider of supercomputing solutions, empowering hundreds of university labs, research organizations, startups, and tinkerers to access ultra-powerful machines. Our mission is to remove the bottleneck of computational resources, enabling scientific progress and innovation.About the RoleWe are seeking an...

  • GPU Software Engineer

    4 weeks ago


    San Diego, California, United States Qualcomm Full time

    About the RoleWe are seeking a highly skilled GPU Software Engineer to join our team at Qualcomm. As a Machine Learning Engineer, you will play a key role in developing software for the latest Machine Learning models on the next generation of devices.Key ResponsibilitiesAccelerate Machine Learning operations for inference and training on the GPUImplement the...


  • San Jose, California, United States Samsung Electronics Full time

    Job SummarySamsung Electronics is seeking a highly skilled Senior GPU Performance Engineer to join our Xclipse GPU software team. As a key member of our team, you will be responsible for developing and optimizing GPU IP from architectural planning to productization, ensuring the highest level of performance and efficiency in our products.Key...


  • San Francisco, California, United States San Francisco Compute Group Full time

    About San Francisco Compute GroupWe are a leading provider of high-performance computing solutions, serving hundreds of university labs, research organizations, startups, and tinkerers. Our mission is to empower scientific progress by providing access to ultra-powerful machines.About the RoleWe are seeking an experienced Machine Learning Systems Engineer to...


  • San Diego, California, United States Samsung Electronics GmbH Full time

    Job Title: Senior Software Engineer, AI and Machine LearningJob Summary:We are seeking a highly skilled Senior Software Engineer to join our AI and Machine Learning team at Samsung Electronics GmbH. As a key member of our team, you will be responsible for designing, developing, and deploying AI and ML models, as well as optimizing their performance on our...


  • San Francisco, California, United States Kuzco Full time

    About KuzcoWe are a cutting-edge technology company specializing in the development of innovative AI solutions. Our team of experienced engineers is dedicated to creating high-performance systems that push the boundaries of what is possible with machine learning.We are building a revolutionary distributed LLM inference network that combines idle GPU capacity...


  • San Francisco, California, United States San Francisco Compute Group Full time

    About the RoleWe are seeking a highly skilled Machine Learning Systems Engineer to join our team at San Francisco Compute Company. As a key member of our team, you will be responsible for optimizing the performance of large AI models, working closely with our customers to understand their needs and develop innovative solutions.Key Responsibilities:Design and...


  • San Jose, California, United States AMD Full time

    About the RoleWe are seeking a highly skilled Machine Learning Performance Engineer to join our team at AMD. As a key member of our organization, you will be responsible for optimizing the performance of machine learning workloads on various hardware configurations.Key ResponsibilitiesPerformance Modeling and Analysis: Develop and implement performance...


  • San Jose, California, United States TikTok Full time

    About the RoleWe're seeking a highly skilled Senior Machine Learning Engineer to join our TikTok Search Team. As a key member of our team, you will be responsible for developing and applying cutting-edge machine learning technologies to improve the search experience for hundreds of millions of users globally.Key ResponsibilitiesOptimize search quality to...


  • San Jose, California, United States Hireio, Inc. Full time

    Job OverviewHireio, Inc. is seeking a highly skilled Senior Machine Learning Engineer to join our Search Ads team. As a key member of our team, you will be responsible for developing and optimizing large-scale Ads systems, including relevance models and strategies, Ads algorithms, and NLP capabilities.Key Responsibilities:Participate in the development of a...

  • Senior GPU Engineer

    4 weeks ago


    San Francisco, California, United States Succinct Full time

    About the RoleWe are seeking a highly skilled Senior GPU Engineer to join our team at Succinct, a leading innovator in zero-knowledge proofs and zkVM technology. As a key member of our engineering team, you will play a critical role in designing, developing, and optimizing software solutions to accelerate our zkVM, SP1, and contribute to the development of...


  • San Jose, California, United States Advanced Micro Devices Full time

    About the RoleWe are seeking a highly skilled Principal Machine Learning Performance Optimization Engineer to join our team at Advanced Micro Devices. As a key member of our organization, you will be responsible for optimizing the performance of machine learning workloads on our hardware platforms.Key ResponsibilitiesPerformance Modeling and Analysis:...


  • San Jose, California, United States OKXhas Full time

    About the OpportunityWe are seeking a highly skilled and experienced Senior Machine Learning Engineer to join our Risk Engineering Team at OKX. The ideal candidate will be adept in developing and implementing advanced machine learning models to enhance our capabilities in fraud detection, including bot detection, credit card chargeback prevention, and...


  • San Jose, California, United States PayPal Full time

    Job Title: Senior Software Machine Learning EngineerAt PayPal, we're revolutionizing commerce globally, and we're looking for a talented Senior Software Machine Learning Engineer to join our Consumer Machine Learning team. As a key member of our team, you'll be responsible for designing, developing, and maintaining our recommendation infrastructure on PayPal...


  • San Jose, California, United States TikTok Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our team in pioneering the future of large language model-based applications. As a key member of our team, you will play a critical role in setting the direction and goals for a sizable team, in terms of project impact, ML system design, and ML...


  • San Jose, California, United States TikTok Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our Search Team at TikTok. As a key member of our team, you will be responsible for developing and applying cutting-edge machine learning technologies to improve the search experience for hundreds of millions of users globally.Key ResponsibilitiesOptimize search quality to...


  • San Jose, California, United States TikTok Full time

    About the RoleWe're seeking a highly skilled Senior Machine Learning Engineer to join our Search Team at TikTok. As a key member of our team, you'll be responsible for developing and applying cutting-edge machine learning technologies to improve the search experience for hundreds of millions of users globally.Key ResponsibilitiesOptimize search quality to...