Staff Engineer, AI/ML Software Compiler

5 Months ago • 8-10 Years • Research & Development • $153,075 PA - $236,925 PA

Job Summary

Job Description

The AGI Computing Lab at Samsung Semiconductor seeks a Staff Engineer specializing in AI/ML software compilers. Responsibilities include designing and implementing ML compilers for high-performance deep learning applications, optimizing compilers for various hardware platforms, designing a staged lowering infrastructure, optimizing data locality for energy efficiency, and collaborating with hardware architects. The role requires strong compiler design knowledge, experience with high-performance computing, and familiarity with frameworks like PyTorch, TensorFlow, or JAX. Troubleshooting, documentation, and staying updated on industry trends are also key aspects. This position is crucial for advancing the capabilities of AI/ML workloads on the company's platforms, contributing to the development of Artificial General Intelligence.
Must have:
  • Compiler design and optimization
  • High-performance computing experience
  • LLVM/MLIR (preferred)
  • PyTorch, TensorFlow, or JAX familiarity
  • Hardware architecture knowledge (CPUs, GPUs, TPUs, NPUs)
Perks:
  • 4+ weeks paid time off
  • Holidays and sick leave
  • Fertility care or adoption stipend
  • Medical travel support
  • Errand service
  • On-demand apps and paid therapy sessions
  • Onsite Café and gym, virtual classes
  • Flexible environment
  • Charitable giving match

Job Details

Please Note:

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

Advancing the World’s Technology Together
Our technology solutions power the tools you use every day--including smartphones, electric vehicles, hyperscale data centers, IoT devices, and so much more. Here, you’ll have an opportunity to be part of a global leader whose innovative 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 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 10+ years of working experiences in silicon development or MS in Computer/Electrical Engineering or Computer Science with 8+ years of relevant working experience or PhD and 5+ 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-VL1

What We Offer
The pay range below is for all roles at this level across all US locations and functions. Individual pay rates depend on a number of factors—including the role’s function and location, as well as the individual’s knowledge, skills, experience, education, and training. We also offer incentive opportunities that reward employees based on individual and company performance. 

This is in addition to our diverse package of benefits centered around the wellbeing of our employees and their loved ones. In addition to the usual Medical/Dental/Vision/401k, our inclusive rewards plan empowers our people to care for their whole selves. An investment in your future is an investment in ours.

Give Back With a charitable giving match and frequent opportunities to get involved, we take an active role in supporting the community.
Enjoy Time Away You’ll start with 4+ weeks of paid time off a year, plus holidays and sick leave, to rest and recharge.
Care for Family Whatever family means to you, we want to support you along the way—including a stipend for fertility care or adoption, medical travel support, and an errand service.
Prioritize Emotional Wellness With on-demand apps and paid therapy sessions, you’ll have support no matter where you are.
Stay Fit Eating well and being active are important parts of a healthy life. Our onsite Café and gym, plus virtual classes, make it easier.
Embrace Flexibility Benefits are best when you have the space to use them. That’s why we facilitate a flexible environment so you can find the right balance for you.

Base Pay Range

$153,075 - $236,925 USD

Equal Opportunity Employment Policy 

Samsung Semiconductor takes pride in being an equal opportunity workplace dedicated to fostering an environment where all individuals feel valued and empowered to excel, regardless of race, religion, color, age, disability, sex, gender identity, sexual orientation, ancestry, genetic information, marital status, national origin, political affiliation, or veteran status.

When selecting team members, we prioritize talent and qualities such as humility, kindness, and dedication. We extend comprehensive accommodations throughout our recruiting processes for candidates with disabilities, long-term conditions, neurodivergent individuals, or those requiring pregnancy-related support. All candidates scheduled for an interview will receive guidance on requesting accommodations.

Recruiting Agency Policy

We do not accept unsolicited resumes. Only authorized recruitment agencies that have a current and valid agreement with Samsung Semiconductor, Inc. are permitted to submit resumes for any job openings.

Covid-19 Policy
To help keep our employees, customers, and communities safe, we’ve developed guidelines for our teams. Currently, we encourage vaccination for all employees and may require it depending on job functions (e.g., traveling for business, meeting with customers). While visiting our offices or attending team events, we ask employees to complete a daily health questionnaire and complete a weekly COVID test. Our COVID policies are subject to change depending on public health, regulatory and business circumstances. 

Applicant Privacy Policy
https://semiconductor.samsung.com/us/careers/privacy

 

Similar Jobs

Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Seattle, Washington, United States (On-Site)
5 Months ago
Starkflow - Principal Full Stack Developer

Starkflow

Karnataka, India (Hybrid)
1 Month ago
Epic Games - Senior Machine Learning Rendering Engineer

Epic Games

(On-Site)
1 Month ago
NVIDIA - Senior System Software Engineer - AI Performance and Efficiency Tools

NVIDIA

Santa Clara, California, United States (Hybrid)
1 Month ago
GoMotive - Computer Vision Engineer

GoMotive

(Remote)
1 Day ago
Google - Software Engineering Manager, Wearables and Health Data Privacy

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
2 Weeks ago
Krafton  - PUBG IP Franchise China Business PM (Entry-Level to 3 Years)

Krafton

Seoul, South Korea (On-Site)
1 Month ago
Trackman - Machine Learning Developer with DSP experience (Python/C++)

Trackman

Hørsholm, Denmark (On-Site)
1 Month ago
Riot Games - Associate Art Director, Characters - Unpublished R&D Product

Riot Games

Los Angeles, California, United States (On-Site)
5 Months ago
Cadence - Principal Design Engineer

Cadence

Bengaluru, Karnataka, India (On-Site)
7 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Senior Software Engineer, Machine Learning, YouTube

Google

San Bruno, California, United States (On-Site)
4 Months ago
Google - Senior Software Engineer, SDLC, Gemini Code Assist

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
NVIDIA - Senior Solutions Architect - Generative AI

NVIDIA

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Google - Senior Research Engineer, AI/ML

Google

London, England, United Kingdom (On-Site)
2 Days ago
Dolby Laboratories - AIOps Research Scientist

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
The Walt Disney Company - Senior Data Scientist - NLP/LLM

The Walt Disney Company

Glendale, California, United States (On-Site)
2 Weeks ago
GoMotive - Computer Vision Engineer

GoMotive

Pakistan (Remote)
1 Month ago
ByteDance - Research Scientist- Applied Machine learning Graduates (AML) - 2024 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Canva - Machine Learning Engineer Intern

Canva

Auckland, Auckland, New Zealand (Remote)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in San Jose, California, United States

Monsters - 3D Environment Artist

Monsters

Austin, Texas, United States (On-Site)
1 Month ago
MarketScale - Experience Center Host

MarketScale

Dallas, Texas, United States (On-Site)
20 Hours ago
AI Dash - Staff Machine Learning / Perception Engineer

AI Dash

Palo Alto, California, United States (Hybrid)
18 Hours ago
Interactive Brokers - Entry- Level Career Opportunities: Institutional Services

Interactive Brokers

Greenwich, Connecticut, United States (Hybrid)
6 Months ago
Google - Software Developer, Ads Bidding Optimization

Google

New York, New York, United States (On-Site)
2 Days ago
Google - Senior Program Manager I, Legal Operations, Global Affairs

Google

Chicago, Illinois, United States (On-Site)
1 Week ago
NVIDIA - Senior Software Engineer, RTL Optimization Tools

NVIDIA

Santa Clara, California, United States (On-Site)
3 Weeks ago
NVIDIA - Senior Software Architect, AI Networking

NVIDIA

Austin, Texas, United States (Remote)
1 Month ago
Mattel  Inc  - Associate Manager, Global Brand Marketing - Barbie

Mattel Inc

El Segundo, California, United States (On-Site)
6 Months ago
Google - Software Engineering Manager II, Dash Developer Experience

Google

Portland, Oregon, United States (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

KPIT - Android Middleware Developer/Lead/Architect

KPIT

Bengaluru, Karnataka, India (On-Site)
9 Months ago
Fluence - Lead Engineer - Advanced Battery Modules

Fluence

Houston, Texas, United States (Hybrid)
6 Months ago
Google - ASIC Platform Software Architect, Silicon

Google

Hyderabad, Telangana, India (On-Site)
2 Days ago
Aristocrat Gaming - Manager, Electrical Engineering

Aristocrat Gaming

Las Vegas, Nevada, United States (Hybrid)
2 Months ago
Google - ASIC Engineer, IP Design, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
2 Weeks ago
Fluence - Controls Software Engineer-II(m/f/d)

Fluence

Erlangen, Bavaria, Germany (Hybrid)
6 Months ago
NVIDIA - Deep Learning Engineer, Datacenters

NVIDIA

Bengaluru, Karnataka, India (Hybrid)
4 Weeks ago
Krafton  - Yeoksam Office Help Desk Representative

Krafton

Seoul, South Korea (On-Site)
1 Month ago
Zuru - Computational Mechanics Engineer

Zuru

Modena, Emilia-Romagna, Italy (Hybrid)
6 Months ago
Trackman - Team Lead - Radar & High-Speed Electronics

Trackman

Denmark (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (Hybrid)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

San Jose, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Samsung Semiconductor

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug