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’s ML Workflows organization. ML Workflows includes Artifacts, the versioned data layer for objects logged to W&B; Registry, the organization-wide source of truth for production-ready assets; Automations, the platform’s event and action layer; and Launch, infrastructure for running 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 collaborate with product managers, designers, and ML platform engineers, and speak directly with enterprise ML teams about their workflows. Team placement is based on an engineer’s strengths and interests. As W&B products are built onto CoreWeave infrastructure, the role offers an opportunity 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 engineering coursework.
- At least 6 years of software engineering experience, including leading significant technical initiatives or architectural changes.
- Strong proficiency in Go, or in another compiled language such as C, C++, C#, Rust, or Java, with the ability to ramp up on Go quickly.
- Working proficiency in Python.
- Ability to investigate complex systems you did not write and diagnose and fix difficult bugs across service boundaries.
- Track record of mentoring engineers and raising the technical bar.
- Strong communication skills, including explaining complex technical decisions to different audiences.
- Depth in one or two areas such as 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.
- Experience with containers, Kubernetes, cloud infrastructure, or orchestration.
- Experience with TypeScript, React, complex state management, or frontend performance.
- Experience with schema design, query performance, large-scale storage, or analytical stores such as ClickHouse.
- Experience with API design, GraphQL query optimization, data fetching strategies, SDKs, or client libraries.
- Experience with message queues such as Kafka or PubSub, delivery semantics, idempotency, or 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 as the platform grows.
- Collaborate with product and design to turn complex user requirements into technical implementations.
- Work directly with customers to understand ML workflow challenges and incorporate their 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 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 an employer match.
- Flexible PTO.
- Catered lunch each day at office and data center locations.
- Discretionary bonus and equity awards, 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
