Senior Machine Learning Engineer for GPU Optimization
5 days ago
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 RoleWe'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
- 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
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.
-
Senior Machine Learning Engineer
3 weeks ago
San Jose, California, United States Adobe Full timeUnlock 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...
-
Senior Machine Learning Engineer
1 week ago
San Jose, California, United States QuantumScape Full timeJob 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...
-
Senior GPU Performance Engineer
1 week ago
San Jose, California, United States Samsung Electronics Full timeJob 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...
-
Machine Learning Systems Engineer
4 weeks ago
San Francisco, California, United States San Francisco Compute Group Full timeAbout 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 timeAbout 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...
-
Senior GPU Performance Engineer
2 weeks ago
San Jose, California, United States Samsung Electronics Full timeJob 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...
-
Machine Learning Systems Engineer
4 weeks ago
San Francisco, California, United States San Francisco Compute Group Full timeAbout 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 timeJob 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 timeAbout 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...
-
Machine Learning Systems Engineer
3 weeks ago
San Francisco, California, United States San Francisco Compute Group Full timeAbout 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 timeAbout 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...
-
Senior Machine Learning Engineer
3 weeks ago
San Jose, California, United States TikTok Full timeAbout 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...
-
Senior Machine Learning Engineer
6 days ago
San Jose, California, United States Hireio, Inc. Full timeJob 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 timeAbout 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 timeAbout 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:...
-
Senior Machine Learning Engineer
4 weeks ago
San Jose, California, United States OKXhas Full timeAbout 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...
-
Senior Software Machine Learning Engineer
6 days ago
San Jose, California, United States PayPal Full timeJob 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 timeAbout 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...
-
Senior Machine Learning Engineer
2 weeks ago
San Jose, California, United States TikTok Full timeAbout 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 Machine Learning Engineer
2 weeks ago
San Jose, California, United States TikTok Full timeAbout 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...