Create Email Alert

Email Alert for

ⓘ There was an unexpected error processing your request.

Please refresh the page and try again.

If the problem persists, please contact us with your issue.

Email address is already registered

You can always manage your preferences and update your interests to ensure you receive the most relevant opportunities.

Would you like to [visit your alert settings] now?

Success! You're now signed up for Job Alerts

Get ready to discover your next great opportunity.

Similar Jobs

  • Samsung Electronics GmbH

    Principal Engineer, AI/ML Software Compiler

    San Jose, CA, United States

    • Ending Soon

    Principal Engineer, AI/ML Software Compiler Job Title Principal Engineer, AI/ML Software Compiler Job Location SSI San Jose Main Office Category Engineering - Software Job Type Full-time Job # 41938 Advancing the World’s Technology Together Our technology solutions power the tools you use every day--including smartphones, electric vehicle

    Job Source: Samsung Electronics GmbH
  • Conductor

    Staff Engineer, AI/ML Software Compiler

    San Jose, CA, United States

    What You’ll Do The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads whil

    Job Source: Conductor
  • Conductor

    Senior Engineer, AI/ML Software Compiler

    San Jose, CA, United States

    What You’ll Do The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads whil

    Job Source: Conductor
  • Etched

    ML Compiler Frontend Engineer

    Cupertino, CA, United States

    • Ending Soon

    ML Compiler Frontend Engineer Etched is building the hardware for superintelligence. GPUs and TPUs are flexible AI chips that can run many kinds of models: CNNs, RNNs, LSTMs, and more. But today, almost all AI workloads, from ChatGPT to self-driving cars, are done on one model architecture: transformers. Using flexible AI chips for transformers is

    Job Source: Etched
  • Etched

    ML Compiler Backend Engineer

    Cupertino, CA, United States

    • Ending Soon

    5+ years of experience writing production-grade software Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals Able to write production-grade code in C++ and in Python Experience with modern compiler IRs, including at least one of (LLVM, MLIR, Relay) Experience with PyTorch Responsibilities Des

    Job Source: Etched
  • Google

    ML Compiler Software Engineer, TPU Horizontal Scaling

    Sunnyvale, CA, United States

    Minimum qualifications: Bachelor’s degree or equivalent practical experience. 2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree. 2 years of experience with data structures or algorithms. Preferred qualifications: Master's degree or PhD in Computer Science or rela

    Job Source: Google
  • Meta

    Software Engineer, Systems ML - Frameworks / Compilers / Kernels_

    Menlo Park

    **Summary:** In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectur

    Job Source: Meta
  • Meta

    Software Engineer, Systems ML - Frameworks / Compilers / Kernels_

    Menlo Park

    **Summary:** In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectur

    Job Source: Meta

Principal Engineer, AI/ML Software Compiler

San Jose, CA, United States

What You’ll Do

The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving the complex system-level challenges posed by the growing demands of future AI/ML workloads. Our team is committed to designing and developing scalable platforms that can effectively handle the computational and memory requirements of these workloads while minimizing energy consumption and maximizing performance. To achieve this goal, we collaborate closely with both hardware and software engineers to identify and address the unique challenges posed by AI/ML workloads and to explore new computing abstractions that can provide a better balance between the hardware and software components of our systems. Additionally, we continuously conduct research and development in emerging technologies and trends across memory, computing, interconnect, and AI/ML, ensuring that our platforms are always equipped to handle the most demanding workloads of the future. By working together as a dedicated and passionate team, we aim to revolutionize the way AI/ML applications are deployed and executed, ultimately contributing to the advancement of AGI in an affordable and sustainable manner. Join us in our passion to shape the future of computing!

Location: Hybrid, working onsite at our office 3 days per week with the flexibility to work remotely the remainder of your time

Reports to: VP

Design and implement ML compilers for high-performance deep learning applications.

Optimize compilers for efficient execution of deep learning models on various hardware platforms.

Design a staged lowering infrastructure to meet rapidly evolving workload requirements effectively.

Design an algorithm to optimize data locality to minimize energy consumption.

Work closely with hardware architects and developers to integrate new ML techniques and algorithms into the compiler.

Collaborate with cross-functional teams to define and deliver ML compiler features and improvements.

Troubleshoot and debug compiler issues, and provide technical support to customers.

Contribute to the development of ML compiler documentation and user guides.

Stay up-to-date with the latest trends and advancements in the field of ML compilers and hardware.

What You Bring

BS in Computer/Electrical Engineering or Computer Science with 20+ years of working experiences in silicon development or MS in Computer/Electrical Engineering or Computer Science with 18+ years of relevant working experience or PhD and 15+ years of relevant working experience preferred.

Strong background in compiler design and optimization techniques.

Experiences in developing and optimizing software for high-performance computing systems

Experiences in LLVM / MLIR (preferred)

Familiarity with PyTorch, Tensorflow, or JAX.

Familiarity with hardware architectures such as CPUs, GPUs, TPUs, and NPUs.

Strong analytical and problem-solving skills

Excellent communication and interpersonal skills

Ability to work independently and as part of a team

You’re inclusive, adapting your style to the situation and diverse global norms of our people.

An avid learner, you approach challenges with curiosity and resilience, seeking data to help build understanding.

You’re collaborative, building relationships, humbly offering support and openly welcoming approaches.

Innovative and creative, you proactively explore new ideas and adapt quickly to change.

#LI-MD1

#J-18808-Ljbffr

Apply

Create Email Alert

Create Email Alert

Email Alert for Principal Engineer, AI/ML Software Compiler jobs in San Jose, CA, United States

ⓘ There was an unexpected error processing your request.

Please refresh the page and try again.

If the problem persists, please contact us with your issue.

Email address is already registered

You can always manage your preferences and update your interests to ensure you receive the most relevant opportunities.

Would you like to [visit your alert settings] now?

Success! You're now signed up for Job Alerts

Get ready to discover your next great opportunity.