Staff Engineer, AI/ML Software Compiler

Job expired!

Join Samsung Semiconductor, Inc. as a Staff Engineer, AI/ML Software Compiler

Please Note:
To provide the best candidate experience with high application volumes, we limit applications to a total of 10 over 6 months.

Advancing the World’s Technology Together

Our innovative technology solutions power the tools you use every day, including smartphones, electric vehicles, hyperscale data centers, IoT devices, and more. At Samsung Semiconductor, Inc., you’ll have the opportunity to be part of a global leader whose pioneering designs are pushing the boundaries of what’s possible and powering the future.

We believe innovation and growth are driven by an inclusive culture and a diverse workforce. We’re dedicated to empowering people to be their true selves. Together, we’re building a better tomorrow for our employees, customers, partners, and communities.

What You’ll Do

The AGI (Artificial General Intelligence) Computing Lab is dedicated to solving complex system-level challenges posed by the advancing AI/ML workloads. Our team designs scalable platforms to meet computational and memory requirements efficiently, while minimizing energy consumption and maximizing performance. Collaborating with both hardware and software engineers, we identify and address unique challenges, exploring new computing abstractions to balance hardware and software components.

  • 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 effectively meet evolving workload requirements.
  • Optimize data locality algorithms to minimize energy consumption.
  • Work closely with hardware architects and developers to integrate new ML techniques and algorithms.
  • Collaborate with cross-functional teams to define and deliver ML compiler features and improvements.
  • Troubleshoot and debug compiler issues, providing technical support to customers.
  • Contribute to the development of ML compiler documentation and user guides.
  • Stay updated with the latest trends and advancements in ML compilers and hardware.

What You Bring

  • BS in Computer/Electrical Engineering or Computer Science with 10+ years of experience, or MS in Computer/Electrical Engineering or Computer Science with 8+ years of experience, or PhD with 5+ years of relevant working experience preferred.
  • Strong background in compiler design and optimization techniques.
  • Experience in developing and optimizing software for high-performance computing systems.
  • Experience in LLVM / MLIR (preferred).
  • Familiarity with PyTorch, TensorFlow, or JAX.
  • Understanding of 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.
  • Adaptability to diverse global norms and inclusive behavior.
  • Curiosity and resilience with a data-driven approach to challenges.
  • Collaborative mindset to build relationships, offer support, and welcome new approaches.
  • Innovative and creative, with quick adaptability to change.

#LI-VL1

What We Offer

The pay range below applies to all roles at this level across all US locations and functions. Individual pay rates depend on factors such as function, location, experience, and education. We offer incentive opportunities based on personal and company performance, along with a diverse benefits package centered on employee well-being.

  • Charitable giving match and frequent community involvement opportunities.
  • 4+ weeks of paid time off per year, plus holidays and sick leave.
  • Support for family needs, including fertility care/adoption stipends, medical travel support, and errand services.
  • Emotional wellness support with on-demand apps and paid therapy sessions.
  • Onsite Café and gym, plus virtual fitness classes to encourage a healthy lifestyle.
  • Flexible work environment to balance personal and professional life.

Base Pay Range: $153,075—$236,925 USD

Equal Opportunity Employment Policy

Samsung Semiconductor takes pride in being an equal opportunity workplace, fostering an