Senior Software Engineer, ML Workflows - Weights & Biases

CoreWeave is hiring a Senior Software Engineer for the Weights & Biases ML Workflows team. The role owns the architecture, development, and production reliability of platform systems that support moving models and data through W&B, partnering across backend, SDK, frontend, product, and customer teams.

Responsibilities

• Own the architecture and evolution of a core platform area, designing systems that scale to billions of artifacts and high-volume event and job throughput
• Dive deep into system architecture to find optimization opportunities, solve complex bugs, and make the difficult calls that balance short-term delivery against long-term platform health
• Work across the stack as the problem requires, 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, keeping the experience seamless for the customers who depend on them
• 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, and bring that feedback into the product cycle
• Mentor engineers and raise the technical bar across the team

Skills

• Bachelor's degree or foreign equivalent in Computer Science, Computer Engineering, Data Science, Information Systems, or a closely related technical or quantitative field involving substantial coursework in Engineering
• 6+ years of software engineering experience, including time leading significant technical initiatives or architectural changes
• Strong proficiency in Go, or in another compiled language (C, C++, C#, Rust, Java) with the ability to ramp on Go quickly
• Working proficiency in Python
• Comfort diving into complex systems you did not write, to diagnose and fix difficult bugs across service boundaries
• A track record of mentoring engineers and raising the technical bar around you
• Strong communication, including the ability to explain complex technical decisions to different audiences
• Distributed systems and infrastructure. Designing and operating services in production, containers and Kubernetes, cloud infrastructure, orchestration
• Frontend and full-stack. TypeScript and React, complex state management, frontend performance work, taking a feature all the way through to the UI
• Data at scale. Schema design, query performance, large-scale storage, and analytical stores such as ClickHouse
• APIs and developer surfaces. API design, GraphQL query optimization and data fetching strategies, SDKs and client libraries
• Event-driven systems. Message queues such as Kafka or PubSub, delivery semantics, idempotency, and retries
• (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
• Experience in ML infrastructure, MLOps, or developer tools is a plus

Benefits

• Discretionary bonus, based on eligibility
• Equity awards, based on eligibility
• Medical, dental, and vision insurance — 100% paid for by CoreWeave
• Company-paid life insurance
• Voluntary supplemental life insurance
• Short- and long-term disability insurance
• Flexible Spending Account
• Health Savings Account
• Tuition reimbursement
• Ability to participate in the Employee Stock Purchase Program (ESPP)
• Mental wellness benefits through Spring Health
• Family-forming support provided by Carrot
• Paid parental leave
• Flexible, full-service childcare support with Kinside
• 401(k) with a generous employer match
• Flexible PTO
• Catered lunch each day in our office and data center locations
• A casual work environment
• A work culture focused on innovative disruption

Company Overview

Company H1B Sponsorship