Job Summary:
SambaNova is a company focused on generative AI solutions, and they are seeking an ML Features Solutions Engineer to drive the development and optimization of core ML features for enterprise deployment. This role combines deep ML expertise with hands-on engineering to deliver production-grade capabilities to customers.
Responsibilities:
โข Design and implement core ML features including model optimization, quantization, and inference enhancements
โข Optimize model performance for latency, throughput, and memory efficiency on SambaNova hardware
โข Develop and improve features such as Function Calling, Structured Output, and JSON mode conformance
โข Create end-to-end ML solutions that showcase platform capabilities and accelerate customer adoption
โข Convert cutting-edge ML research into practical, deployable product features
โข Establish benchmarks and quality standards for ML features in production environments
โข Work with SDK team to ensure ML features are properly exposed and documented for developers
โข Support enterprise customers implementing advanced ML features in their workflows
โข Partner with ML research, platform engineering, and customer teams
Qualifications:
Required:
โข Masterโs degree or higher in Computer Science, Machine Learning, Electrical Engineering, or related field
โข 5+ years of industry experience in ML engineering or applied ML research
โข 3+ years of hands-on experience with large language models and transformer architectures
โข Expert proficiency in Python and deep learning frameworks: PyTorch (required), TensorFlow, or JAX
โข Experience with model optimization techniques: quantization, pruning, distillation, efficient inference
โข Strong understanding of LLM inference optimization: KV cache, batching strategies, memory management
โข Experience deploying ML models to production at scale
โข Track record of translating research concepts into production features
Preferred:
โข PhD in Machine Learning, NLP, or related field
โข Experience with custom hardware acceleration (TPUs, custom ASICs)
โข Hands-on experience with inference frameworks: vLLM, TensorRT-LLM, or similar
โข Experience with function calling and tool use in LLMs
โข Knowledge of structured generation and constrained decoding
โข Experience with ML feature development in enterprise contexts
โข Contributions to open-source ML projects
Company:
SambaNova is a complete AI platform that delivers AI inference and scalable solutions, providing a seamless integration for agentic AI. Founded in 2017, the company is headquartered in Palo Alto, USA, with a team of 201-500 employees. The company is currently Growth Stage.