As a Staff Engineer, you will scale metrics and storage systems, lead technical direction, maintain performance, and mentor junior engineers.
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
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
Top Skills
AWS
Azure
Bigtable
Clickhouse
GCP
Go
Kafka
Kubernetes
MySQL
Postgres
Pub/Sub
Python
Terraform
Typescript
Weights & Biases San Francisco, California, USA Office
1479 Folsom St, San Francisco, CA, United States, 94103
Similar Jobs
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
As a Senior Linux System Admin, you'll manage cloud infrastructure, develop automation tools, troubleshoot services, and improve operational processes.
Top Skills:
AnsibleBashCactiCentosF5IcingaJavaScriptLinuxLogstashMariadbMySQLNagiosNginxOraclePuppetPythonRedhatSplunk
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
The Senior Software Quality Engineer ensures quality of AI-powered search and workflow features on the Now Platform, developing automation tests and collaborating with developers on testing strategies.
Top Skills:
Azure OpenaiEclipseGenerative AiGitHugging FaceJavaJavaScriptJenkinsJunitLangchainMavenOpenaiSeleniumTestng
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Build and optimize a high-performance AI inferencing platform. Utilize Python and Golang, collaborate with teams, ensure model reliability, and stay updated on technologies.
Top Skills:
AIGoMachine LearningPrompt EngineeringPython
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
