Machine Learning Compiler Engineer

1 week ago


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

At Advanced Micro Devices, Inc., we're revolutionizing the world of technology with our cutting-edge products. Our mission is to accelerate next-generation computing experiences, and we're looking for talented individuals to join our team.

Job Description

We're seeking an experienced Machine Learning Compiler Engineer to develop a high-performance compiler targeting AMD Inference Accelerator AIE hardware devices. Your primary responsibility will be to implement and improve passes in the compiler, ensuring optimal code generation for AMD x86 CPU using AMD ZenDNN.

Key Responsibilities:
  • Implement model partitioning in ML Frameworks and/or MLIR
  • Integrate compiler and compiled model with ML Frameworks (such as PyTorch and TensorFlow)
  • Mentor and provide guidance to others
  • Learn latest industry trends and bring new ideas to the team
Requirements:

To succeed in this role, you'll need strong object-oriented programming skills in C/C++ and Python. Experience with ML Compiler and Runtime technologies, such as OneDNN, MLIR, XLA, OpenXLA, IREE, and OpenAI Triton compiler, is highly desirable. Additionally, you should have a deep understanding of PyTorch, TensorFlow, ONNX, JAX, and machine learning concepts.

Salary and Benefits

We offer a competitive salary range of $120,000 - $180,000 per year, depending on your experience and qualifications. You'll also receive comprehensive benefits, including stock options, employee discounts, and access to state-of-the-art facilities.

About This Role

This is a full-time position based in [Location]. As a member of our team, you'll have the opportunity to work on groundbreaking projects, collaborate with talented engineers, and contribute to the development of innovative technologies. If you're passionate about machine learning and eager to take your career to the next level, we encourage you to apply.



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

    We are at the forefront of innovation, driving technological advancements that enrich our industry, communities, and world. Our mission is to build great products that accelerate next-generation computing experiences.Our core team is developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices. The...


  • San Jose, California, United States AMD Full time

    **About AMD**We are a leading technology company dedicated to transforming lives with our innovative products. Our mission is to accelerate next-generation computing experiences and push the limits of innovation to solve the world's most important challenges.**Job Summary**We are seeking an experienced engineer to join our core team working on developing a...


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

    **About the RoleWe are seeking an experienced Software Engineer to join our Ads machine learning data platform team at TikTok. This is a unique opportunity to lead projects and empower the team with technical expertise.As a key member of our team, you will be responsible for building scalable and reliable ML-oriented data infrastructure systems. Your strong...


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

    About the RoleAs a Senior Machine Learning Engineer at TikTok, you will be responsible for providing users with a first-class search experience by building a strong and robust infrastructure and platform to support product fast iteration and key feature development.Key Responsibilities• Support business development of TikTok Search, providing the best...


  • San Jose, California, United States Cadence Design Systems Full time

    Cadence Design Systems is a leading electronic design automation company providing software, hardware, and intellectual property to design advanced semiconductor chips. We are seeking an exceptional Ai Researcher, Machine Learning Engineer to join our team.About UsWe are proud to be recognized by Fortune Magazine as one of the 100 Best Companies to Work For....


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

    Job DescriptionWe are seeking a highly skilled Machine Learning Engineer to join our team in Los Angeles or Singapore. The ideal candidate will have expertise in building industry-leading recommendation systems and delivering end-to-end machine learning solutions.About UsTikTok is the leading destination for short-form mobile video, inspiring creativity and...


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

    About the RoleTikTok is the leading destination for short-form mobile video. In this role, you will be part of the team that makes TikTok possible. Key Responsibilities• Develop highly scalable machine learning systems and state-of-the-art machine learning models to improve ads ranking quality and optimize advertisers' marketing strategies. Examples...


  • San Francisco, California, United States Cadre Full time

    Job DescriptionWe are seeking a highly skilled Machine Learning Engineer to join our team at Cadre. As a key member of our data science and engineering team, you will play a critical role in developing and implementing advanced machine learning models and algorithms to drive business growth and improve user experiences.Key ResponsibilitiesCollaborate with...


  • San Jose, California, United States Adobe Full time

    About AdobeAt Adobe, we're dedicated to revolutionizing digital experiences for everyone. With a passion for empowering creators and innovators, we provide the tools and expertise needed to design and deliver exceptional digital experiences.We're committed to creating an inclusive and collaborative environment where everyone feels valued and respected. Our...


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

    {'Responsibilities': 'Key ResponsibilitiesWork with our world-class engineers to build content understanding models for content moderation.- Work on state-of-the-art content understanding techniques including but not limited to unimodal/multimodal content classifiers, graph models, representation learning models, and LLM.- Collaborate with product teams to...


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

    About the RoleThe Multimedia Arch team at TikTok is responsible for user experience optimization and cost optimization for the whole video on demand and live streaming system. As a machine learning engineer, you will collaborate with the product and engineering teams to identify opportunities and improve overall system performance and efficiency by...


  • San Jose, California, United States GLO Comms Full time

    Transformative Engineering Manager - AI InnovationGLO Comms, a tech-forward company in Culver City, is seeking an experienced Engineering Manager specializing in Machine Learning Models to drive innovation and lead a dynamic team.This permanent position offers opportunities for growth while working with some of the brightest minds shaping tomorrow's...


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

    Job OverviewWe are seeking a Senior Machine Learning Engineer to join our dynamic Brand Ads team at Hireio, Inc. This role presents an exciting opportunity to shape the future of monetization on TikTok by developing cutting-edge machine learning models that drive brand awareness, engagement, and revenue.The ideal candidate will have expertise in designing...


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

    Search Ads Optimization EngineerAbout the Role:We are seeking a highly skilled Senior Machine Learning Engineer to join our team at Hireio, Inc., focusing on search ads optimization. As part of our team, you will work on machine learning models for ads relevance, NLP for query understanding, and optimizing ad delivery at scale.Key Responsibilities:Develop...


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

    About the OpportunityWe are seeking a skilled Senior Machine Learning Engineer to join our Search Ads team at Hireio, Inc. This role offers a unique chance to shape the future of search monetization systems and work on cutting-edge technologies in ads, machine learning, NLP, ranking, and more.Job ResponsibilitiesDevelop high-impact ads relevance models and...


  • San Francisco, California, United States Primer Full time

    About Primer: Primer is a leading AI software company that empowers critical organizations with trusted decision-ready AI.The Role: As a Machine Learning Engineer at Primer, you will lead the development and improvement of our data-driven products, from research to production. You will leverage your expertise in data science, algorithms, and language models...


  • San Francisco, California, United States Adobe Full time

    About the Role:We're seeking a talented Machine Learning Engineer to join our team at Adobe, where we're pushing the boundaries of AI innovation.Key Responsibilities:Develop and deploy machine learning models in production environments.Collaborate with cross-functional teams to integrate AI-driven solutions into our products.Design and implement data...


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

    About the Role: We are seeking a talented Machine Learning Engineer to join our Knowledge Graph team. As a key member of the team, you will be responsible for the construction and development of vast knowledge graphs that encapsulate real-world products, aiding in feed ranking, recommendations, and ads.Key Responsibilities:Craft intricate knowledge graphs...


  • San Jose, California, United States ZipRecruiter Full time

    Job OverviewWe are seeking a seasoned Tech Lead Machine Learning Engineer to join our Brand Ads Team, where you will have the opportunity to build and manage systems with high performance, scalability, and availability.


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

    Job Title: Machine Learning and AI EngineerDescription:We are seeking a highly skilled Machine Learning and AI Engineer to join our team at Hireio, Inc. In this role, you will be responsible for exploring and developing large-scale language models and optimizing enterprise applications.Key Responsibilities:1. Large-Scale Model Development:Design, develop,...