ML Features Solutions Engineer

Home Jobs ML Features Solutions Engineer S ML Features Solutions Engineer SambaNova On-site Austin, Texas, United States; San Jose, California, United States Full-time 30 days ago Apply on SambaNova Product SW 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. About The Role We are seeking an ML Features Solutions Engineer to join our Product and Solution Engineering team, driving the development and optimization of core ML features for enterprise deployment. This role combines deep ML expertise with hands-on engineering, working at the intersection of ML research and product development to deliver production-grade capabilities to our customers. Role This role is critical for accelerating ML feature development and bridging the gap between ML research and product engineering and will be driving the following: Core ML Feature Development: Drive improvements to ML features including model optimization, inference performance, and feature enhancements.Production-Ready Solutions: Build and deploy production-ready ML solutions for enterprise customers with focus on reliability and scale.Research to Product Bridge: Translate ML research innovations into practical product features and customer-facing capabilities.Cross-Team Collaboration: Work closely with SDK, testing, and customer teams to ensure ML features meet enterprise requirements.Impact: Accelerates ML feature development and optimization, enabling faster time-to-market for new capabilities while ensuring enterprise-grade quality and performance. Responsibilities Design and implement core ML features including model optimization, quantization, and inference enhancementsOptimize model performance for latency, throughput, and memory efficiency on SambaNova hardwareDevelop and improve features such as Function Calling, Structured Output, and JSON mode conformanceCreate end-to-end ML solutions that showcase platform capabilities and accelerate customer adoptionConvert cutting-edge ML research into practical, deployable product featuresEstablish benchmarks and quality standards for ML features in production environmentsWork with SDK team to ensure ML features are properly exposed and documented for developersSupport enterprise customers implementing advanced ML features in their workflowsPartner with ML research, platform engineering, and customer teams Required Qualifications Master’s degree or higher in Computer Science, Machine Learning, Electrical Engineering, or related field5+ years of industry experience in ML engineering or applied ML research3+ years of hands-on experience with large language models and transformer architecturesExpert proficiency in Python and deep learning frameworks: PyTorch (required), TensorFlow, or JAXExperience with model optimization techniques: quantization, pruning, distillation, efficient inferenceStrong understanding of LLM inference optimization: KV cache, batching strategies, memory managementExperience deploying ML models to production at scaleTrack record of translating research concepts into production features Preferred Qualifications PhD in Machine Learning, NLP, or related fieldExperience with custom hardware acceleration (TPUs, custom ASICs)Hands-on experience with inference frameworks: vLLM, TensorRT-LLM, or similarExperience with function calling and tool use in LLMsKnowledge of structured generation and constrained decodingExperience with ML feature development in enterprise contextsContributions to open-source ML projects What We Offer Work on cutting-edge ML features powering the fastest AI inference platformDirect impact on product capabilities used by enterprise customers globallyCollaborate with world-class ML researchers and engineersBay Area location enabling close collaboration with core ML teamsCompetitive compensation and benefitsOpportunity to shape the future of enterprise AI Base Salary Range Base Pay Range$200,000—$270,000 USD 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. 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. Benefits Summary For US-Based, Full-Time Employment Positions 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. Interested in this role? Applications are handled on SambaNova's site. Apply Now Related Jobs View all AI Engineering S Senior Product Engineer, Growth & Lifecycle Infrastructure - Music & Audio Stability AI Remote Full-time T AI Researcher, Core ML (Turbo) Together AI On-site Full-time C Account Solution Architect CoreWeave On-site Full-time T Backend Software Engineer - Data Platform & AI Data Products Together AI On-site Full-time S AI Applications Ops Lead, GPS Scale AI On-site Full-time P Member of Technical Staff (Machine Learning Engineer, Search) Perplexity Remote Full-time Back to all jobs Related Jobs View all AI Engineering S Senior Product Engineer, Growth & Lifecycle Infrastructure - Music & Audio Stability AI Remote Full-time T AI Researcher, Core ML (Turbo) Together AI On-site Full-time C Account Solution Architect CoreWeave On-site Full-time T Backend Software Engineer - Data Platform & AI Data Products Together AI On-site Full-time S AI Applications Ops Lead, GPS Scale AI On-site Full-time P Member of Technical Staff (Machine Learning Engineer, Search) Perplexity Remote Full-time