Software Engineer

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.

Weave is our new lightweight toolkit for tracking and evaluating GenAI applications. Our users are relying on Weave to understand and improve their AI-powered applications. They’re on the cutting edge, building Agents, Retrieval Augmented Generation (RAG) systems, and more.

We’re looking for a software engineer to help build Weave. You’ll play a foundational role on this small but growing team, developing a new generation of tools for the most consequential emerging market today. You’ll ship core functionality, gather feedback directly from early users, and use your own intuition to continually improve the product.

If you have a strong interest in GenAI, love building and shipping quickly, and enjoy talking to users, this is the perfect role for you.

What you’ll achieve (Responsibilities)

• Implement robust and ergonomic open-source client APIs.
• Design and implement integrations with popular LLM tools such as LlamaIndex, LangSmith, and Instructor.
• Scale and improve our backend performance to support low latency and high volume production use cases.
• Build polished and delightful features such as multi-modal evaluations, LLM playgrounds, and dataset editing/drill-down capabilities.
• Actively meet with and iterate based on customer feedback, understanding their workflows and addressing key bugs.

What we’re looking for (Requirements)

• Bachelor’s degree in Computer Science, or related field.
• 4 years or more of prior experience in Software Engineering.
• Deep knowledge of Python and/or TypeScript.
• Experience (personal or professional) building GenAI-powered applications.
• Ability to operate effectively amid ambiguity and adapt rapidly to a changing business environment. Previous startup experience a plus.
• Preferred: Experience managing large-scale online production services or OLAP databases.

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
• 💵 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
• Explore benefits by country

$177,000 - $245,000 a year

The US base pay for this position ranges from $177,000 USD per year in our lowest geographic market up to $245,000 USD per year in our highest geographic market. Weights & Biases is committed to providing competitive salary, equity and benefits packages for all full-time employees. Individual compensation will be commensurate with the candidate's experience, qualifications, and geographic location.

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