Senior Staff Software Engineer

6 days ago


San Jose, California, United States AMD Full time
Transforming Lives with AMD Technology

We are passionate about using technology to enrich our industry, communities, and the world. Our mission is to build innovative products that accelerate next-generation computing experiences, from the data center to PCs, gaming, and embedded systems. At AMD, we push the limits of innovation to solve the world's most pressing challenges, striving for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.

The Role

We are seeking a world-class AI frameworks and compiler engineer to provide technical leadership in the development of various AI frameworks in the AMD ecosystem. As a key member of our team, you will drive technical direction for next-generation frameworks for AI model training and inference on a wide variety of AMD devices, including MI Instinct, Radeon GPUs, XDNA devices, and datacenter CPUs such as EPYC. You will work to enhance AI framework capabilities to enable cutting-edge models on AMD's cutting-edge hardware.

The Person

If you are passionate about AI/ML frameworks, software architecture, and/or compilers, this is your opportunity to join our team. You will work in one of the core areas, such as AI/ML frameworks components (e.g. PyTorch, TensorFlow, ONNX/OnnxRuntime), AI compiler and runtime, and/or optimization tooling to accelerate AI/ML workloads on AMD hardware platforms. You will collaborate closely with AI researchers to drive the development of framework components to efficiently map AI models to run on a variety of HW AI accelerators. You should be someone who can work in a dynamic, fast-paced development environment, with excellent leadership and collaboration skills.

Requirements
  • Experience with development in one of the focus areas: AI frameworks, AI compiler stack, and/or performance tuning and optimizations for workloads running on ML accelerator HWs.
  • Experience with ML frameworks such as PyTorch, OnnxRuntime, JAX, TensorFlow.
  • Proficient in C++ programming.
  • Experience developing and debugging in Python.
  • Excellent skills at designing Python tools and libraries used by large numbers of users.
  • Experience with AI model architectures, e.g. Transformers, CNNs.
  • Team player and ready to work with a geographically distributed team.
Preferred Experience
  • Highly preferred - knowledge of custom accelerator hardware.
  • Depth of experience with AI compilers, dataflow graphs, MLIR, and dialects, compiler optimizations, hardware-specific optimizations.
  • Experience with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus.
  • Depth of experience with model optimization, including quantization, sparsity.
  • Solid understanding of transformer architecture, LLMs, MoE.
  • Experience on LLMs/LMMs finetuning, distillation, and/or RLHF.
Academic Credentials

BS, MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or related technical fields.

Location

San Jose, CA



  • San Jose, California, United States PayPal Full time

    Job Title: Senior Staff Software EngineerAt PayPal, we're committed to building an equitable and inclusive global economy. As a Senior Staff Software Engineer, you'll play a critical role in shaping the future of our consumer-facing applications. We're looking for a talented engineer with a passion for mobile development and a strong track record of...


  • San Jose, California, United States PayPal Full time

    About the RoleWe are seeking a highly skilled Senior Staff Software Engineer to join our team at PayPal. As a Senior Staff Software Engineer, you will be responsible for leading the development of complex software systems and mentoring junior engineers. You will work closely with cross-functional teams to design, develop, and deploy high-quality software...


  • San Jose, California, United States BILL Full time

    Join Our Team as a Senior Staff Software EngineerBILL is a leader in financial automation software for small and mid-size businesses (SMBs). As a champion of SMBs, we are dedicated to automating the future of finance so businesses can thrive. Hundreds of thousands of businesses trust BILL solutions to manage financial workflows, including payables,...


  • San Jose, California, United States Triunity Software Full time

    Job DescriptionWe are seeking a highly skilled Senior Validation Engineer to join our team at Triunity Software, Inc. as a key member of our Product Development and Staff Augmentation services.About the Role:Process and equipment qualification (IQ/OQ/PQ)Development of test methods for Design V&V, Process Validation, and ProductionValidation of test...


  • San Jose, California, United States SENIOR SPIRIT OF ROSELLE PARK Full time

    About the RoleWe are seeking an experienced Senior Staff Engineer to join our Backend team. As a key member of our engineering organization, you will be responsible for designing and building large-scale applications, collaborating with cross-functional teams, and driving technical innovation.Key ResponsibilitiesDesign and develop complex, multi-tiered...


  • San Jose, California, United States Triunity Software Full time

    Job Title: Senior Validation EngineerWe are seeking a highly skilled Senior Validation Engineer to join our team at Triunity Software, Inc. As a Senior Validation Engineer, you will be responsible for ensuring the quality and reliability of our medical devices by developing and implementing validation plans, protocols, and procedures.Key...


  • San Jose, California, United States NIO Full time

    About NIONIO is a pioneering company in the premium smart electric vehicle market, founded in 2014. Our mission is to shape a joyful lifestyle by building a community that shares joy and grows together with users.Our TechnologyWe design, develop, and manufacture premium smart electric vehicles, driving innovations in next-generation technologies such as...


  • San Jose, California, United States SENIOR SPIRIT OF ROSELLE PARK Full time

    About the RoleWe are seeking an experienced Senior Software Architect to join our team at SENIOR SPIRIT OF ROSELLE PARK. As a key member of our engineering team, you will be responsible for designing and building large-scale applications that meet the needs of our customers.Key ResponsibilitiesDesign and develop complex, multi-tiered software applications...


  • San Jose, California, United States Rootshell Inc Full time

    Job Title: Senior Software EngineerWe are seeking a highly skilled Senior Software Engineer to join our team at Rootshell Inc. As a Senior Software Engineer, you will be responsible for designing, implementing, and testing software applications.Key Responsibilities:Design and develop software applications using various programming languages and...


  • San Jose, California, United States MediaTek Full time

    Job SummaryWe are seeking a highly skilled Senior Software Engineer/Staff Software Engineer to join our team at MediaTek. As a key member of our software development team, you will be responsible for designing, developing, and delivering high-quality software solutions that meet the needs of our customers.Key ResponsibilitiesCollaborate with Customers and...


  • San Jose, California, United States Software AG Full time

    Job SummaryWe are seeking a highly skilled and experienced software engineer to join our Software AG CONNX team in a remote capacity. The successful candidate will be responsible for designing, implementing, and maintaining features for the CONNX product suite, which is primarily written in cross-platform C++.Key ResponsibilitiesLead and own the design,...


  • San Carlos, California, United States Elegen Full time

    Job Title: Senior Staff Software EngineerElegen is a pioneering company in the field of synthetic biology, leveraging cutting-edge technologies to deliver innovative DNA synthesis solutions. As a Senior Staff Software Engineer, you will play a crucial role in building and maintaining software tools that advance the manufacture of genes.Key...


  • San Jose, California, United States Triunity Software Full time

    Job Title: Validation EngineerWe are seeking a highly skilled Validation Engineer to join our team at Triunity Software, Inc. as a Senior Recruiter.The ideal candidate will have a strong background in medical device manufacturing, with a minimum of 3 years of experience in test method validation and/or measurement system analysis.The successful candidate...


  • San Jose, California, United States TEKsystems Full time

    Job Title: Senior Software EngineerWe are seeking a highly skilled Senior Software Engineer to join our team at TEKsystems. As a key member of our engineering team, you will be responsible for designing, developing, and deploying high-quality software solutions that meet the needs of our clients.Responsibilities:Design and implement software solutions using...


  • San Jose, California, United States Rootshell Enterprise Technologies Inc. Full time

    Job Title: Senior Software EngineerWe are seeking a highly skilled Senior Software Engineer to join our team at Rootshell Enterprise Technologies Inc.Key Responsibilities:Design, implement, and run automated tests with a test frameworkPick and ramp up with new technologies/languages quicklyDetermine the cause of automated test failuresDocument and complete...


  • San Jose, California, United States ANSYS Full time

    Job Title: Senior Application EngineerAt Ansys, we are seeking a highly skilled Senior Application Engineer to join our team. As a key member of our technical support team, you will be responsible for providing expert-level support to our customers, helping them to overcome complex technical challenges and achieve their goals.Key Responsibilities:Develop and...


  • San Jose, California, United States Lynx Software Technologies Full time

    Job Title: Sr. Software Engineer, Customer SuccessJob Summary:We are seeking a highly skilled Sr. Software Engineer to join our Customer Success team. As a key technical resource, you will ensure the successful deployment and support of our Lynx product implementations. This is a customer-facing role that requires excellent communication skills and...


  • San Mateo, California, United States GEICO Full time

    Job Title: Senior Staff EngineerGEICO is seeking an experienced Senior Staff Engineer to join our Platform as a Service (PaaS) Messaging Team. As a key member of our team, you will be responsible for designing and implementing highly scalable distributed platforms that will help transform GEICO's insurance business.Job SummaryWe are looking for a talented...


  • San Diego, California, United States ServiceNow Full time

    Job Title: Senior Staff Software Engineer, PlatformXAt ServiceNow, we're pushing the boundaries of what's possible with AI and machine learning. We're looking for a talented Senior Staff Software Engineer to join our PlatformX team and help us bring innovative solutions to our customers.About the RoleWe're seeking a highly experienced software engineer with...


  • San Jose, California, United States Supermicro Full time

    Job Title: Senior Software EngineerSupermicro is seeking a highly skilled Senior Software Engineer to join our team in San Jose, California. As a critical member of our system department, you will be responsible for implementing AI/ML applications to support organizational initiatives.Key Responsibilities:Design, develop, test, and deploy applications using...