Senior ML Platform Engineer: Scale AI Workflows

Coreweave; Sunnyvale (CA); On-site; USD 165,000 - 220,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401(k) with employer match

Flexible PTO

Tuition Reimbursement

Paid Parental Leave

Job summary

CoreWeave, a leading AI cloud provider, seeks a Senior Software Engineer to own the architecture of ML workflows within the Weights & Biases platform. You will drive end-to-end design across the Go backend, GraphQL, Python SDK, and TypeScript frontend, collaborating with product, design, and ML teams to scale systems and deliver reliable AI tooling.

You’ll mentor engineers, tackle complex technical challenges, and help shape the future of a unified AI platform built on CoreWeave infrastructure.

Qualifications

• Bachelor’s degree or foreign equivalent in Computer Science, Computer Engineering, Data Science, Information Systems, or a closely related field.
• 6+ years of software engineering experience including leading significant initiatives or architectural changes.
• Strong proficiency in Go, or in another compiled language with ramp on Go.
• Working proficiency in Python.
• Comfort diagnosing complex systems across service boundaries.
• Track record of mentoring engineers and raising the technical bar.
• Strong communication, including explaining complex decisions to diverse audiences.

Responsibilities

• Own architecture and evolution of a core platform area, scalable to billions of artifacts and high-volume throughput.
• Dive deep into system architecture to optimize, fix bugs, and make long-term health decisions.
• Work across Go backend, GraphQL APIs, Python SDK, and TypeScript frontend.
• Lead migrations and re-architecture of live systems, ensuring seamless customer experience.
• Refactor technical debt to improve maintainability and developer productivity.
• Establish engineering patterns and best practices for scaling the platform.
• Collaborate with product and design to translate complex requirements into clean implementations.
• Engage with customers to understand ML workflow challenges and incorporate feedback.
• Mentor engineers and raise the technical bar across the team.

Skills

Go programming

Python

Distributed systems

Mentoring

Communication

Architectural leadership

Education

Bachelor’s degree in Computer Science / Computer Engineering / Data Science / Information Systems

Tools

GraphQL

TypeScript

Kubernetes

Job description

CoreWeave, a leading AI cloud provider, seeks a Senior Software Engineer to own the architecture of ML workflows within the Weights & Biases platform. You will drive end-to-end design across the Go backend, GraphQL, Python SDK, and TypeScript frontend, collaborating with product, design, and ML teams to scale systems and deliver reliable AI tooling.

You’ll mentor engineers, tackle complex technical challenges, and help shape the future of a unified AI platform built on CoreWeave infrastructure.