At Weights & Biases, our mission is to build the best tools for AI developers. We founded our company on the insight that while there were excellent tools for developers to build better code, there were no similarly great tools to help ML practitioners build better models. Starting with our first experiment tracking product, we have since expanded our solution into a comprehensive AI developer platform for organizations focused on building their own deep learning models and generative AI applications.
Weights & Biases is a Series C company with $250M in funding and over 200 employees. We proudly serve over 1,000 customers and more than 30 foundation model builders including customers such as OpenAI, NVIDIA, Microsoft, and Toyota.
Reporting to the Engineering Manager, this Senior/Staff Software Engineer - Registry Platform will own the technical architecture and evolution of our Registry product. This role will lead the critical transition from our legacy Model Registry to a scalable, organization-wide Global Registry platform, while also architecting for a future that supports the extensive needs of large foundation model builders and customers focused on ML model fine-tuning, compliance, system-of-record requirements, and comprehensive monitoring capabilities.
This Senior/Staff Software Engineer will partner closely with product managers, ML platform engineers, and the design team to build and implement a robust, performant registry system that meets the growing demands of enterprise ML teams while maintaining a seamless user experience during the transition.
Responsibilities:
Lead the architectural transformation of our Registry platform, designing systems that scale to support millions/billions of artifacts for our customers.Dive deep into system architecture to uncover optimization opportunities, solve complex bugs, and make difficult technical decisions that balance short-term fixes with long-term platform health.Identify and refactor areas of technical debt, improving code maintainability and developer productivity.Mentor a team of 3-4 Software Engineers through complex technical challenges, elevating their capabilities through code reviews and design guidance.Establish engineering patterns and best practices that balance immediate feature delivery with long-term platform health.Collaborate with product and design to translate complex user requirements into elegant technical implementations.Interface directly with customers to understand their ML workflow challenges, gathering feedback and incorporating it into the product development cycle.Practice agile methodologies with users at the center, maintaining a closed feedback loop from customer need to technical implementation to validation.Requirements:
7+ years of software engineering experience, with 3+ years leading significant technical initiatives or architectural changesStrong proficiency in PythonStrong proficiency in at least one compiled language (C++, C, C#, Go), with Go experience being a significant bonusExpert-level TypeScript knowledge with demonstrated ability to implement deep frontend optimizations and complex state managementAdvanced experience with GraphQL APIs, query optimization, and sophisticated data fetching strategiesDemonstrated success improving performance and maintainability of complex web applicationsTrack record of mentoring engineers and elevating team capabilities through technical leadershipComfort diving into complex systems to diagnose and solve difficult bugs across the full stackExceptional communication skills with ability to explain complex technical concepts to various audiencesBenefits: