Senior Software Engineer, ML Workflows - Weights & Biases
About The Position
The Senior Software Engineer will own the architecture and evolution of core systems in the Weights & Biases platform, reporting to the Senior Engineering Manager for ML Workflows. ML Workflows includes Artifacts, the versioned data layer for objects logged to W&B; Registry, an organization-wide source of truth for production-ready assets; Automations, the platform’s event and action layer; and Launch, execution infrastructure for customer ML and agent workloads. The role owns systems end to end, from design through production reliability, across the W&B backend, Python SDK, and frontend. The engineer will work closely with product managers, designers, and ML platform engineers, and directly with enterprise ML teams to understand their workflows. Engineers are matched to a team based on their strengths and interests. The products are being built onto CoreWeave infrastructure, offering room to shape that integration.
Requirements
- 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.
- At least 6 years of software engineering experience, including time leading significant technical initiatives or architectural changes.
- Strong proficiency in Go, or proficiency in another compiled language (C, C++, C#, Rust, or Java) and ability to ramp up on Go quickly.
- Working proficiency in Python.
- Ability to dive into complex systems not previously written by the candidate and diagnose and fix difficult bugs across service boundaries.
- Track record of mentoring engineers and raising the technical bar.
- Strong communication skills, including ability to explain complex technical decisions to different audiences.
- Depth in at least one or two of the following areas, with range to work beyond them: distributed systems and infrastructure; frontend and full-stack development; data at scale; APIs and developer surfaces; or event-driven systems.
Nice To Haves
- Experience in ML infrastructure, MLOps, or developer tools.
- Relevant depth may include production services, containers and Kubernetes, cloud infrastructure, orchestration, TypeScript and React, complex state management, frontend performance, schema design, query performance, large-scale storage, analytical stores such as ClickHouse, API design, GraphQL query optimization, SDKs and client libraries, or message queues such as Kafka or PubSub and their delivery semantics, idempotency, and retries.
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.
- Analyze system architecture to identify optimization opportunities, solve complex bugs, and balance short-term delivery with long-term platform health.
- Work across Go backend services and GraphQL APIs, the Python SDK, and the TypeScript frontend as needed.
- Lead migrations and re-architecture of systems carrying live customer traffic while maintaining a seamless customer experience.
- Identify and refactor technical debt to improve maintainability and developer productivity.
- Establish engineering patterns and best practices that support platform growth.
- Collaborate with product and design to turn complex user requirements into technical implementations.
- Work directly with customers to understand their ML workflow challenges and bring feedback into the product cycle.
- Mentor engineers and raise the technical bar across the team.
Benefits
- Medical, dental, and vision insurance, 100% paid for by CoreWeave for eligible US-based full-time employees.
- 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 at office and data center locations.
- Discretionary bonus and equity awards, based on eligibility.
- Comprehensive benefits program, based on eligibility.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
