Senior Software Engineer, ML Workflows - Weights & Biases

Job Description

As a Senior Software Engineer, ML Workflows at Weights & Biases, part of CoreWeave, you will own the architecture and evolution of core systems in the Weights & Biases platform.

About the Role

You will report to the Senior Engineering Manager for ML Workflows and work on ML Workflows, which owns what customers use to move models and data through W&B. This includes Artifacts, Registry, Automations, and Launch.

Responsibilities

• Design and evolve systems that scale to billions of artifacts and high-volume event and job throughput.
• Optimize system architecture, solve complex bugs, and make decisions that balance short-term delivery against long-term platform health.
• Work across the stack, from Go backend services and GraphQL APIs to the Python SDK and the TypeScript frontend.
• Lead migrations and re-architecture of systems carrying live customer traffic.
• Identify and refactor technical debt, improving maintainability and developer productivity.
• Establish engineering patterns and best practices that hold up as the platform grows.
• Collaborate with product and design to turn complex user requirements into clean technical implementations.
• Work directly with customers to understand their ML workflow challenges.
• Mentor engineers and raise the technical bar across the team.

Requirements

• Bachelor’s degree or foreign equivalent in Computer Science, Computer Engineering, Data Science, Information Systems, or a closely related technical or quantitative field.
• 6+ years of software engineering experience, including time leading significant technical initiatives or architectural changes.
• Strong proficiency in Go, or in another compiled language with the ability to ramp on Go quickly.
• Working proficiency in Python.
• Comfort diving into complex systems to diagnose and fix difficult bugs.
• A track record of mentoring engineers and raising the technical bar around you.
• Strong communication skills.

Nice to Have

• Experience in distributed systems and infrastructure.
• Frontend and full-stack experience with TypeScript and React.
• Data at scale, including schema design and query performance.
• APIs and developer surfaces, including API design and GraphQL query optimization.
• Event-driven systems, including message queues like Kafka or PubSub.
• Experience in ML infrastructure, MLOps, or developer tools.

What We Offer

• Base pay ranging from $165,000 to $220,000.
• Comprehensive benefits package, including medical, dental, and vision insurance, company-paid Life Insurance, short and long-term disability insurance, and more.
• Flexible PTO, catered lunch, and a casual work environment.