Staff Software Engineer - Metrics

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 proudly serves over 1,000 customers and more than 30 foundation model builders including customers such as OpenAI, NVIDIA, Microsoft, and Toyota.

As a Staff Engineer, you'll lead the effort to scale our metrics and storage systems, ensuring they meet the complex demands of our most advanced customers. You’ll play an instrumental role in the evolution of our platform as we grow our capability to ingest and query petabytes of data, making critical technical decisions that optimize the performance, reliability, and cost-effectiveness of our systems.

You will set the technical direction for the team, guiding the organization to balance short-term deliverables with strategic, long-term architectural improvements. You'll partner closely with product management, revenue teams, and other engineering groups to shape and deliver the future of W&B’s flagship Models product, supporting experiment tracking and analytics utilized by over 2,500 leading machine learning and AI teams worldwide.

Responsibilities:

• Design and implement infrastructure that is scalable, efficient, and tailored to customer needs.

• Lead the maintenance and monitoring of existing services, identifying and executing necessary improvements to ensure ongoing performance and reliability.

• Participate in team-wide rotations to respond to customer support issues and site outages.

• Communicate and collaborate effectively with internal and external stakeholders to achieve optimal outcomes.

• Lead and mentor junior engineers, supporting their professional growth and development within the company.

Requirements:

• 8+ years of experience in software engineering, with a focus on data platforms and/or distributed systems.

• Strong software engineering fundamentals and proficiency in at least one modern programming language (e.g., Python, Go, Typescript).

• Extensive experience designing and scaling customer-facing APIs in production environments, ideally leveraging systems like MySQL, Postgres, Clickhouse, Bigtable, Pub/Sub, Kafka, etc.

• Hands-on experience with Kubernetes, Terraform, and major cloud providers (e.g., GCP, AWS, Azure).

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)

• Supplemental benefits may be available depending on your location

• Explore benefits by country

$210,000 - $290,000 a year

This position has an annual estimated salary of $210,000 - $290,000 per year. Weights & Biases is committed to providing competitive salary, equity and benefits packages for all full-time employees. The actual pay may vary depending on your skills, qualifications, experience, and work location.

Export Control Compliance

This position requires access to export controlled information. To conform to U.S. Government export regulations applicable to that information, applicant must either be (A) a U.S. person, defined as a (i) U.S. citizen or national, (ii) U.S. lawful permanent resident (green card holder), (iii) refugee under 8 U.S.C. § 1157, or (iv) asylee under 8 U.S.C. § 1158, (B) eligible to access the export controlled information without a required export authorization, or (C) eligible and reasonably likely to obtain the required export authorization from the applicable U.S. government agency. Weights & Biases may, for legitimate business reasons, decline to pursue any export licensing process.

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