Staff Engineer, AI/ML Software Compiler jobs in United States
cer-icon
Apply on Employer Site
company-logo

Samsung Electronics · 11 hours ago

Staff Engineer, AI/ML Software Compiler

Samsung Electronics is a global leader in technology solutions, and they are seeking a Staff Engineer for their AI/ML Software Compiler team. The role involves designing and implementing ML compilers for high-performance deep learning applications while collaborating with hardware architects and cross-functional teams to optimize compiler features and address workload challenges.

Consumer ElectronicsElectronicsManufacturingMobile AppsMobile Devices
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

Compiler designOptimization techniquesHigh-performance computingLLVM / MLIRPyTorchTensorFlowHardware architecturesAnalytical skillsCuriosityResilienceProblem-solving skillsCommunication skillsInterpersonal skillsTeamworkCollaborationCreativity

Required

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
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

Preferred

Experiences in LLVM / MLIR

Benefits

Medical/Dental/Vision/401k
Charitable giving match
4+ weeks of paid time off a year
A stipend for fertility care or adoption
Medical travel support
Virtual vet care for your fur babies
On-demand apps and free confidential therapy sessions
Onsite Café and gym
Virtual classes
Flexible environment

Company

Samsung Electronics

company-logo
Samsung Electronics is a technology company that engages in consumer electronics, IT and mobile communications, and device solutions.

H1B Sponsorship

Samsung Electronics has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (206)
2024 (143)
2023 (234)
2022 (169)
2021 (144)
2020 (144)

Funding

Current Stage
Public Company
Total Funding
$4.75B
Key Investors
U.S. Department of Commerce
2024-04-15Grant· $4.75B
2023-01-10Post Ipo Equity
1975-06-11IPO

Leadership Team

leader-logo
Boo-Keun Yoon
President & CEO
leader-logo
Jong Hee Han
Vice Chairman & CEO
Company data provided by crunchbase