Current jobs related to Staff Applied Machine Learning Software Engineer – Generative AI - San Jose, California - AMD


  • San Jose, California, United States Tik Tok Full time

    Job DescriptionTikTok is a leading destination for short-form mobile video, and our mission is to inspire creativity and bring joy. We're looking for a talented Machine Learning Engineer to join our Generative AI team under Monetization Technology.As a Machine Learning Engineer, you will be responsible for developing cutting-edge Generative AI technologies...


  • San Francisco, California, United States Relyance AI Full time

    Job SummaryRelyance AI is seeking a highly skilled Senior Software Engineer, Machine Learning to join our team. As a key member of our engineering team, you will be responsible for designing, developing, and deploying machine learning models and algorithms to drive business outcomes.Key ResponsibilitiesDevelop and implement machine learning models and...


  • San Francisco, California, United States Scale AI Full time

    Join Scale AI as a Research EngineerWe are seeking a highly skilled Research Engineer to join our team at Scale AI. As a Research Engineer, you will play a key role in designing and developing next-generation data pipelines and supervision strategies for large language models.Key Responsibilities:Collaborate with researchers from top foundation model labs to...


  • 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 TikTok Full time

    Role SummaryAs a Senior Machine Learning Engineer on our Generative AI team, you will be responsible for developing cutting-edge AI technologies that drive creative efficiency for advertisers, agencies, and creators. You will work closely with global teams to design and implement scalable and reliable software infrastructure that meets the needs of our...

  • Software Engineer

    4 weeks ago


    San Jose, California, United States Coactive AI Full time

    Unlock the Power of AI with CoactiveCoactive is revolutionizing the way businesses harness the power of machine learning to unlock the potential of unstructured data. As a Software Engineer focused on AI solutions, you will be a key contributor to our Solutions team, bridging sales, customer success, product, and engineering.Your primary goal is to ensure...


  • San Jose, California, United States Tik Tok Full time

    Job Title: Senior Machine Learning Engineer - Generative AIAbout the Role:We are seeking a highly skilled Senior Machine Learning Engineer to join our Generative AI team under Monetization Technology. As a key member of our team, you will be responsible for developing cutting-edge Generative AI technologies across various modalities, including text, image,...


  • San Jose, California, United States Adobe Full time

    Our MissionAt Adobe, we're passionate about empowering people to create beautiful and powerful digital experiences. We're seeking a highly skilled Generative AI/Machine Learning Engineer and Data Scientist to join our AI/ML/DS team. The ideal candidate will have extensive experience with traditional machine learning techniques, as well as recent work...


  • San Jose, California, United States Tik Tok Full time

    Job Title: Senior Machine Learning EngineerJob Summary:We are seeking a highly skilled Senior Machine Learning Engineer to join our Generative AI team at TikTok. As a key member of our team, you will be responsible for developing and maintaining efficient, low-latency data pipelines, designing and implementing robust, scalable systems for data curation and...

  • Software Engineer

    2 weeks ago


    San Jose, California, United States Coactive AI Full time

    Unlock the power of visual data with Coactive AI.As a Software Engineer on our AI Applications team, you will play a pivotal role in developing and maintaining RESTful microservices using Python and FastAPI.Leveraging our advanced Multimodal AI Platform (MAP), you'll bridge the gap between customer success, product development, and engineering to deliver...

  • Software Engineer

    3 weeks ago


    San Jose, California, United States Coactive AI Full time

    Coactive is revolutionizing the way businesses harness the power of machine learning to unlock the potential of unstructured data. We are seeking a highly skilled Software Engineer to join our Solutions team as an AI Solutions Expert.About the Role:As an AI Solutions Expert, you will be responsible for delivering AI-focused technical solutions with clear...


  • San Diego, California, United States ServiceNow Full time

    Job Title: Staff Machine Learning EngineerAt ServiceNow, we're seeking a highly skilled Staff Machine Learning Engineer to join our global team. As a key member of our AI/ML solutions team, you'll be responsible for designing and developing cutting-edge cloud-based AI/ML solutions to power intelligent services.Key Responsibilities:Design and develop...


  • San Francisco, California, United States Together AI Full time

    Together AI is seeking an experienced MLOps engineer to develop and deploy scalable AI/ML systems. The ideal candidate will have a strong understanding of machine learning, particularly large language models, and experience with DevOps practices like CI/CD, automation, and containerization.Key ResponsibilitiesDesign and implement runtime systems for...


  • San Francisco, California, United States Scale AI Full time

    Job SummaryThe role of a Machine Learning Engineer at Scale involves bringing cutting-edge techniques in computer vision, deep learning, and natural language processing into a production environment to enhance our products and customer experience. Our research engineers leverage our unique access to massive datasets to deliver improvements to our...


  • San Jose, California, United States Zscaler Full time

    About ZscalerZscaler is a leading cloud security company that provides a comprehensive security platform to protect enterprises from cyber threats. As a Senior Software Engineer, Machine Learning, you will play a critical role in building and scaling our cloud security platform.Our team is responsible for designing and developing large-scale distributed...


  • San Jose, California, United States Adobe Full time

    About the OpportunityWe're seeking a talented Machine Learning Engineer to join our team at Adobe Firefly's applied research organization. As a key member of our team, you'll play a crucial role in enhancing customer experiences through AI and generative technologies.ResponsibilitiesContribute to research and advanced development that can be applied to Adobe...


  • San Diego, California, United States ServiceNow Full time

    Transforming the Way People WorkAt ServiceNow, we're revolutionizing the way organizations operate. As a Staff Machine Learning Engineer, you'll be part of a global team building cutting-edge AI/ML solutions that enhance the employee experience.Key Responsibilities:Design and develop world-class cloud-based AI/ML solutions to power intelligent...


  • San Francisco, California, United States ZIP Full time

    Transforming Business Spend with AIAt Zip, we're revolutionizing the way businesses manage spend with our cutting-edge AI solutions. As a Senior/Staff Machine Learning Engineer, you'll play a crucial role in accelerating our AI product development and improving model accuracy.Key Responsibilities:Design and develop Machine Learning models for AI assistant,...


  • San Diego, California, United States BRAIN CORP Full time

    Job SummaryBRAIN CORP is a leading AI company revolutionizing the retail industry with its cutting-edge technology. We are seeking a highly skilled Senior Software Engineer, Applied Machine Learning to join our Insights team and contribute to the development of innovative products built on our BrainOS platform.As a key member of our team, you will be...


  • San Jose, California, United States Adobe Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our Document Cloud AI team. As a key member of our team, you will be responsible for designing, developing, and deploying cutting-edge AI and machine learning solutions that drive business growth and customer satisfaction.Key Responsibilities:Design and develop AI and...

Staff Applied Machine Learning Software Engineer – Generative AI

2 months ago


San Jose, California, United States AMD Full time
Unlock the Power of Generative AI with AMD

At AMD, we're pushing the boundaries of innovation to solve the world's most important challenges. We're seeking a talented and motivated AI Software Engineer to join our team and contribute to the development of cutting-edge Generative AI solutions.

The Role:

As an AI Software Engineer, you'll be responsible for exploring and improving state-of-the-art research in Generative AI inference solutions. You'll work on developing tools and techniques for model analysis, profiling, and performance projections, as well as architecting and prototyping custom kernels on GPUs and CPUs. Your work will shape and influence AMD's inference and training solutions across the software stack, from frameworks like PyTorch/ONNX to custom hardware like NPU/GPU.

The Person:

We're looking for someone with a deep understanding of foundational AI model architectures, software development, and AI/ML frameworks. You'll have demonstrated efficiently mapping AI models onto GPUs and/or other hardware accelerators. You'll be someone who can work in a dynamic, fast-paced development environment with excellent collaboration skills and a spirit of innovation.

Key Responsibilities:
  • Accelerate inference of Generative AI on AMD's products.
  • Develop tools and techniques for model analysis, profiling, and performance projections.
  • Architect and prototype custom kernels on GPUs and CPUs.
  • Optimize deep learning inference pipeline, including graph compilation using AMD AI compilers.
  • Reproduce and improve upon state-of-the-art quantization, pruning, and optimization algorithms in PyTorch and Python.
  • Develop high-quality software to enable next-gen solutions.
  • Collaborate with AI/ML frameworks and infrastructure teams to enable new algorithms in the platforms.
  • Collaborate with AMD Research and Architecture teams to improve future products.
Preferred Experience:
  • Solid understanding of CNN, Transformer model architectures, LLMs, and Stable Diffusion models.
  • Strong development and debugging skills in Python.
  • Experience with model optimization, including quantization and sparsity algorithms.
  • Experience in development with ML frameworks like PyTorch, ONNXRuntime, JAX, and TensorFlow.
  • Familiarity with AI compilers and execution runtime.
  • Experience in C++ programming for GPUs and/or custom accelerators.
Academic Credentials:
  • BS, MS, or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related technical fields.

At AMD, your base pay is one part of your total rewards package. Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role, such as an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD's Employee Stock Purchase Plan. You'll also be eligible for competitive benefits described in more detail here.