Senior Software Engineer, Models

As a Senior Software Engineer at Weights & Biases, you will have the opportunity to design and build our generative AI platform, powering a new wave of LLM-based applications. You will work directly with users, leveraging cutting‑edge techniques such as Agentic Workflows, Retrieval Augmented Generation (RAG), and more.

This role involves working on an early‑stage product and being a major contributor to its growth and adoption. If you thrive in a fast‑paced startup environment, enjoy iterating closely with users, and want to be on the frontlines of the GenAI revolution, this is the perfect role for you!

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.

Requirements

• Deep knowledge of Python or TypeScript.
• Experience with open‑source ML libraries; contributions to a GenAI/LLM framework are a bonus.
• Experience managing large‑scale online production services or OLAP databases.
• A minimum of 4 years of software engineering experience.
• Ability to operate effectively amid ambiguity and adapt rapidly to a changing business environment.

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

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 View email address on click.appcast.io.