[Hiring] Senior Software Engineer, Registry @Wandb

Wandb; United States; Remote; USD 120,000 - 180,000; Full time

Job summary

Join Wandb as a Senior Software Engineer for the Registry platform, where you'll take charge of architectural transformation, mentor a talented team, and innovate solutions for ML workflows. Our tools empower top-tier machine learning teams, and your expertise will shape the future of AI development. With a flexible remote-first culture, competitive salary, and comprehensive benefits, we invite diverse candidates to apply and thrive in our collaborative environment.

Benefits

Flexible time off

Medical, Dental, and Vision for employees and Family Coverage

Home office budget with a new high-powered laptop

12 weeks of Parental leave

401(k)

Qualifications

• 7+ years of software engineering experience.
• 3+ years leading significant technical initiatives.
• Strong proficiency in Python and TypeScript.

Responsibilities

• Lead the architectural transformation of the Registry platform.
• Mentor a team of 3-4 Software Engineers.
• Identify and refactor areas of technical debt.

Skills

Python

C++

C

C#

Go

TypeScript

GraphQL

Communication

Job description

Mar 19, 2025 - Wandb is hiring a remote Senior Software Engineer, Registry. Salary: competitive salary and equity. Location: USA.

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 ourRegistry platform, designing systems that scale to support millions/billionsof 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 areasof technical debt, improving codemaintainability and developer productivity.

Mentor a team of3-4 Software Engineers through complextechnical challenges, elevating their capabilities throughcode reviews and design guidance.

Establish engineeringpatterns and best practices that balanceimmediate feature delivery with long-term platformhealth.

Collaborate with productand design to translate complex user requirementsinto elegant technical implementations.

Interface directly with customersto understand their ML workflow challenges, gathering feedback and incorporating it into theproduct development cycle.

Practice agile methodologies with users at the center, maintaining a closed feedback loop from customer need to technicalimplementation to validation.

Requirements:

7+ years of software engineering experience, with3+ years leading significant technical initiatives or architectural changes

Strong proficiency in Python

Strong proficiency in atleast one compiled language (C++, C, C#, Go), with Go experiencebeing a significant bonus

Expert-level TypeScript knowledgewith demonstrated ability to implement deepfrontend optimizations and complex state management

Advanced experience with GraphQL APIs, query optimization, and sophisticated datafetching strategies

Demonstrated success improving performance and maintainability of complex web applications

Trackrecord of mentoring engineers and elevating team capabilitiesthrough technical leadership

Comfort diving into complex systems to diagnose and solve difficult bugs acrossthe full stack

Exceptionalcommunication skills with ability to explain complex technical concepts tovarious audiences

Benefits:

️ Flexible time off

????Medical, Dental, and Vision for employees and Family Coverage

Remote first culture with in-office flexibility in San Francisco

Home office budget with a new high-powered laptop

Truly competitive salary and equity

12 weeks of Parental leave (U.S. specific)

401(k) (U.S. specific)

Supplemental benefits may be available depending on your location

Why join us?

Top-tier machine learning teams rely on our tools for their daily work at companies including OpenAI, Toyota Research Institute, Lyft, Samsung, and Pandora.

You'll never stop learning. This role gives you first-hand experience talking with leading researchers in the field, understanding their problems, and directly shaping the product direction.

Our experienced founding team has successfully built and sold ML tools in the past at Figure Eight, and their deep knowledge of our industry, empathy for our users, and skillful management is driving W&B to success.

Customers genuinely benefit from our tool. Here's a quote from Wojciech Zaremba, Cofounder and Robotics Lead, OpenAI: "W&B allows to scale up insights from a single researcher to the entire team, and from a single machine to hundreds of them."

We encourage you to apply even if your experience doesn't perfectly align with the job description as we seek out diverse and creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish with us. We are an equal opportunity employer and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. If you need additional accommodations to feel comfortable during your interview process, reach out [e-mail].

#LI-Remote