Description
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 changes
- Strong proficiency in Python
- Strong proficiency in at least one compiled language (C++, C, C#, Go), with Go experience being a significant bonus
- Expert-level TypeScript knowledge with demonstrated ability to implement deep frontend optimizations and complex state management
- Advanced experience with GraphQL APIs, query optimization, and sophisticated data fetching strategies
- Demonstrated success improving performance and maintainability of complex web applications
- Track record of mentoring engineers and elevating team capabilities through technical leadership
- Comfort diving into complex systems to diagnose and solve difficult bugs across the full stack
- Exceptional communication skills with ability to explain complex technical concepts to various 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 at careers@wandb.com.
#LI-Remote
WeightsBiases
Artificial Intelligence
Data Visualization
Developer Tools
Machine Learning
0 applies
22 views
Other Jobs from WeightsBiases
AI Customer Support Engineer, Tier II - Tokyo (Hybrid)
Remote
Tokyo, Japan
AI Engineer - India (Remote)
Remote
New Delhi, India
Software Engineer, Engineering Productivity (Remote)
Remote
San Francisco, CA
Senior Software Engineer, Sweeps - US (Remote)
Remote
San Francisco, CA
Senior AI Solutions Engineer, Pre Sales - EMEA (Remote)
Remote
London, UK
Similar Jobs
Senior Product Security Engineer
Bengaluru, India
Software Engineer 5 - Partner Engineering
Los Gatos, CA
US
FlightAware - Senior Software Engineer, Backend (Onsite)
Houston, TX
US
Comcast Cybersecurity: Engineer 5, Solution Architect and Integration Engineer
Remote
Philadelphia, PA
Data Engineer
Remote Hybrid
There are more than 50,000 engineering jobs:
Subscribe to membership and unlock all jobs
Engineering Jobs
60,000+ jobs from 4,500+ well-funded companies
Updated Daily
New jobs are added every day as companies post them
Refined Search
Use filters like skill, location, etc to narrow results
Become a member
🥳🥳🥳 452 happy customers and counting...
Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.
To try it out
For active job seekers
For those who are passive looking
Cancel anytime
Frequently Asked Questions
- We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
- We've got about 70,000 jobs from 5,000 vetted companies. No fake or sleazy jobs here!
- We aggregate jobs from 5,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
- We're the only job board *for* software engineers, *by* software engineers… in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. 🛠️
- Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. 🚀
- Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. 🎯
- Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. 📅
What Fellow Engineers Say
