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.
As a Senior Software Engineer on our Artifacts, Registry, and Automations Team, you'll be part of a cutting-edge team that is revolutionizing machine learning workflows. You'll play a crucial role in building and optimizing systems that power artifact management, model versioning, and workflow automation. As a key member, your contributions will enhance our core artifact handling codebase, driving scalable solutions for managing the entire ML lifecycle—from creation and tracking to deployment.
You will have the unique opportunity to influence the development of W&B's emerging products, such as the Registry and Automations. Collaborating closely with across engineering, product, and real-world customers, you'll design impactful features that directly shape the future of machine learning tools. Your work will directly affect user experiences, streamlining artifact management across both cloud and on-prem environments.
Responsibilities:
- Innovate & Impact: Design, develop, and maintain high-performance features for Weights & Biases’ flagship products, ensuring scalable and efficient management of ML workflows and model lifecycle tracking.
- Lead the Charge: Spearhead the development of cutting-edge tools that enable seamless model integration, versioning, and deployment in production environments.
- Collaborate & Create: Work hand-in-hand with product managers, engineering teams, and customers to build robust infrastructure that tackles real-world ML challenges head-on.
- Mentor & Inspire: Contribute to code reviews, provide technical leadership in discussions, and mentor junior engineers to uphold our commitment to coding excellence and best practices.
- User-Centric Development: Champion a user-first approach by integrating customer feedback into product improvements that deliver measurable value.
Requirements:
- 4+ years of experience in software engineering.
- Strong software engineering fundamentals and knowledge of at least 1 modern programming language (Python, Go, Typescript, GraphQL, etc).
- Strong empathy for ML practitioners with a keen interest in understanding and addressing their challenges.
- A desire to be self-driven and find the path to greatest impact.
- Nice to have: experience working with ML Engineers and building tools to allow them to easily connect with their compute.
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."
Our 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)
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 [email protected].
#LI-Remote
Top Skills
What We Do
Weights & Biases helps machine learning teams build better models faster. With a few lines of code, practitioners can instantly debug, compare and reproduce their models — architecture, hyperparameters, git commits, model weights, GPU usage, and even datasets and predictions — and collaborate with their teammates.
