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