Senior Machine Learning Training GPU Optimization Engineer

3 days ago


San Jose, California, United States Adobe Full time

Transforming Digital Experiences with Adobe

We're on a mission to empower creators and innovators to design and deliver exceptional digital experiences. As a Senior Machine Learning Training GPU Optimization Engineer at Adobe, you'll play a critical role in shaping the future of digital experiences.

Key Responsibilities:

• Design and implement efficient forward and backward passes in CUDA/CuTe to optimize machine learning training.

• Develop optimized custom layers in PyTorch to accelerate machine learning training.

• Optimize ML training code for large, distributed training with FP8 to improve performance and efficiency.

• Conduct quality and performance analysis between data types such as BF16 and FP8 for large deep learning models.

• Understand and optimize H100 GPUs to maximize their potential.

• Architect broader, end-to-end optimized training code and schemes with PyTorch for large distributed models.

• Write high-quality, product-level code that is easy to maintain and test following standard methodologies.

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

Compensation:

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. Your recruiter can share more about the specific salary range for the job location during the hiring process.



  • San Jose, California, United States Adobe Full time

    Join Adobe's Team of InnovatorsWe'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 RoleWe're seeking a highly skilled and experienced engineer to...


  • 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 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.Key Responsibilities:Optimize and fine-tune GPU-based systems and applications for maximum...


  • 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 Jose, California, United States Samsung Electronics Full time

    About the RoleSamsung Electronics Co., Ltd. is seeking a highly skilled Senior GPU Performance Engineer to join our team. As a key member of our GPU software team, you will be responsible for designing, developing, and optimizing GPU software to deliver cutting-edge performance and efficiency.Key ResponsibilitiesDesign and develop GPU software to optimize...


  • San Jose, California, United States SAMSUNG Full time

    Position SummarySamsung, a world leader in advanced semiconductor technology, is founded on a simple philosophy – the endless pursuit of excellence will create a better world for all. At Samsung Austin Research and Development Center (SARC) and Advanced Computing Lab (ACL), we are building a center of excellence for Intellectual Property (IP) that is...


  • San Jose, California, United States TikTok Full time

    TikTok's Generative AI TeamWe're a team of innovators at TikTok, dedicated to developing cutting-edge Generative AI technologies that drive creativity and efficiency in the advertising and creative industries.ResponsibilitiesDesign and implement AI-driven applications for text-to-image and text-to-video transformations to support diverse content generation...


  • San Jose, California, United States AMD Full time

    About the RoleWe are seeking a highly skilled Machine Learning Software Engineer to join our AI team at AMD. As a key member of our team, you will be responsible for developing and optimizing Generative AI inference solutions on our products.Key ResponsibilitiesAccelerate inference of Generative AI on AMD's products.Develop tools and techniques for model...


  • 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...


  • San Jose, California, United States Adobe Systems Full time

    Job Title: Senior AI Engineer, PlatformWe are seeking a highly skilled Senior AI Engineer to join our team at Adobe Systems. As a key member of our platform team, you will be responsible for designing, developing, and maintaining robust AI/ML infrastructure solutions to support the training and deployment of large-scale AI models.Key Responsibilities:Design...

  • Senior GPU Engineer

    1 month 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 Hireio, Inc. Full time

    Job OverviewHireio, Inc. is seeking a seasoned Senior Machine Learning Engineer to join our Brand Ads Team. As a key member of our team, you will be responsible for creating innovative monetization products that drive engagement and revenue.Key Responsibilities:Create innovative monetization products that drive engagement and revenue.Participate in the...


  • 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 e-commerce alliance team at TikTok. As a key member of our team, you will be responsible for designing and building large-scale machine learning algorithms to optimize e-commerce search engines.Key ResponsibilitiesDesign and develop machine learning algorithms to...


  • 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...

  • Senior GPU Engineer

    2 weeks ago


    San Francisco, California, United States Succinct Full time

    About SuccinctSuccinct is a pioneering company in the field of zero-knowledge proofs, dedicated to making this complex technology accessible to developers. Our mission is to empower developers to build scalable, interoperable, and private blockchain solutions.The RoleWe are seeking a Senior GPU Engineer to join our team and play a critical role in the...


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

    Transforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled engineer, you'll join a core team of industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The Role:Develop a...

  • Senior GPU Engineer

    4 weeks ago


    San Francisco, California, United States Succinct Full time

    About SuccinctSuccinct is a pioneering company in the field of zero-knowledge proofs, dedicated to making this complex technology accessible to developers. Our mission is to empower developers to build scalable, interoperable, and private blockchain solutions.The RoleWe are seeking a Senior GPU Engineer to join our team and contribute to the development of...