Principal Software Engineer, ML Kernel Optimization

SambaNova Systems

SambaNova Systems

Software Engineering, Data Science
Palo Alto, CA, USA
Posted on Wednesday, February 7, 2024

The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses and operations at scale.

SambaNova Suite™ is the first full-stack, generative AI platform, from chip to model, optimized for enterprise and government organizations. Powered by the intelligent SN40L chip, the SambaNova Suite is a fully integrated platform, delivered on-premises or in the cloud, combined with state-of-the-art open-source models that can be easily and securely fine-tuned using customer data for greater accuracy. Once adapted with customer data, customers retain model ownership in perpetuity, so they can turn generative AI into one of their most valuable assets.

Working at SambaNova

This role presents a unique opportunity to shape the future of AI and the value it can unlock across every aspect of an organization’s business and operations through developing software and hardware that is driving tremendous change across the industry. As a Principal Software Engineer, ML Kernel Optimization, you would be responsible for leading the development and delivery of performance optimized matrix multiply kernels for all of SambaNova’s product lines, and you will work on and coordinate with different layers of the stack. It is a critical role responsible for driving optimizations that enable SOTA machine learning model performance on the SambaNova platform.

Responsibilities

  • Designing and delivering performance optimized kernels for core deep learning operations, like matrix multiply, for various generations of our hardware architecture
  • Collaborate with various teams at SambaNova (hardware and architecture, compiler, and performance) for aiding hardware-software co-design
  • Follow general software engineering best practices including setting up and maintaining regressions
  • Use and aid development of internal tools for simulation
  • Lead a team of engineers focusing on the same mission

Basic Qualifications

  • BS/MS/PhD in CS or EE
  • 7+ years of relevant industry experience
  • Proficiency with C++ and Python

Additional Required Qualifications

  • Along with 7+ years of industry experience, have at least 3 years of experience with relevant deep learning algorithms on accelerators or GPUs
  • Solid understanding of hardware architecture
  • Excellent software programming skills, including debugging and performance analysis
  • Ability to define project goals and scope
  • Good communication and documentation skills

Preferred Qualifications

  • Experience with performance-oriented parallel programming
  • Experience with dataflow architectures

Annual Salary Range and Level

The base salary for this position ranges from $165,000/year up to $210,000/year. This range is based on role, level, and location and reflects the salary target for new hires in the US. Individual pay within the range will depend on a number of factors, including a candidate’s job-related qualifications, skills, competencies and experience, and location.

#LI-SB1

Benefits Summary for US-Based Full-Time Direct Employment Positions

(The Recruiter will provide benefit details for non-US-based roles)

SambaNova offers a competitive total rewards package, including the base salary, plus equity and benefits. We cover 95% premium coverage for employee medical insurance, and 77% premium coverage for dependents and offer a Health Savings Account (HSA) with employer contribution. We also offer Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life, and AD&D insurance plans in addition to Flexible Spending Account (FSA) options like Health Care, Limited Purpose, and Dependent Care. Our library of well-being benefits available to you and your dependents includes a full subscription to Headspace, Gympass+ membership with access to physical gyms, One Medical membership, counseling services with an Employee Assistance Program, and much more.

Submission Guidelines

Please note that in order to be considered an applicant for any position at SambaNova Systems, you must submit an application form for each position for which you believe you are qualified.

If you are a new, recent (within the last two years), or upcoming college graduate and are interested in opportunities with SambaNova Systems, please apply through our university job listings.

EEO Policy

SambaNova Systems is an Equal Opportunity/Affirmative Action Employer. All qualified applicants will receive consideration for employment without regard basis of age (40 and over), color, disability, gender identity, genetic information, marital status, military or veteran status, national origin/ancestry, race, religion, creed, sex (including pregnancy, childbirth, breastfeeding), sexual orientation, and any other applicable status protected by federal, state, or local laws.

Customers turn to SambaNova to quickly deploy state-of-the-art generative AI capabilities within the enterprise. Our purpose-built enterprise-scale AI platform is the technology backbone for the next generation of AI computing.

Headquartered in Palo Alto, California, SambaNova Systems was founded in 2017 by industry luminaries, and hardware and software design experts from Sun/Oracle and Stanford University. Investors include SoftBank Vision Fund 2, funds and accounts managed by BlackRock, Intel Capital, GV, Walden International, Temasek, GIC, Redline Capital, Atlantic Bridge Ventures, Celesta, and several others. Visit us at sambanova.ai. Follow SambaNova Systems on Linkedin.