Jobs for Developers

Software Architect, ML

SambaNovaSystemsFull-time$220k - $270kPalo Alto, CaliforniaMar 16, 2024
Apply for this job

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.

SambaNova is looking for a highly motivated and experienced Software Architect to guide the development of our ML developer workflow. As an AI hardware provider focused on speed and efficiency of models, it is critical to also translate the machine speed and efficiency to the ML developer workflow. Enabling this is a combination of building improved ML Application architecture, Integration with popular tools, smooth integration with compiler and runtime stacks and even optimizing kernels for multiple.

North Star: Reduce the time it takes to develop new ML Models on RDU 

Roles and Responsibilities 

  • Collaborate with multiple ML teams across vision, LLM and speech to architect ML application code that reduces technical debt and time to delivery.
  • Develop and maintain data utilities for various text, image, speech data and ensure flexibility to new data.
  • Automate verification and testing of ML models on RDU.
  • Architect, develop and maintain evaluation benchmark infrastructure for ML models across modalities.
  • Coordinate ML release process with compiler, runtime and other complementary software team releases to ensure smooth production delivery.
  • Set software standards (testing, code structure, gotchas, error handling, oncall) for ML teams working with complementary software teams; lead by example and codify these into code and documentation.


  • 10+ years experience contributing to large codebases and architecting software infrastructure
  • 10+ years experience as tech lead unblocking engineers by diving into code 
  • Strong and principled software and code review practices
  • 5+ years experience with open-source ML Frameworks, e.g., PyTorch, Tensorflow
  • 2+ years experience building engineering tools to enable iterative ML Research and Development
  • Familiarity with design decisions behind common developer tools in the community (e.g. HuggingFace, FastAI, Pytorch Lightning, Mosaic ML
  • Experience developing, training and large scale ML models (LLMs, LVMs, CV, etc) is preferred.
  • Proven track record of managing complex projects across different teams/orgs.
  • Proven success in hiring, onboarding, performance management, goal setting and related processes for the team.
  • B.S, M.S. or PhD in Computer Science or a related field

Annual Salary Range and Level

The base salary for this position ranges from $220,000/year up to $270,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.


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.