Senior ML Workflows Architect: Scale AI Pipelines

Weights & Biases

New York (NY)

On-site

USD 180,000 - 240,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Medical Insurance
Dental Insurance
Vision Insurance
Company-paid Life Insurance
Voluntary Supplemental Life Insurance
Short and Long-term Disability
Flexible Spending Account
Health Savings Account
Tuition Reimbursement
Employee Stock Purchase Program (ESPP)
Mental Wellness Benefits
Family-Forming Support
Paid Parental Leave
Childcare Support
401(k) with Employer Match
Flexible PTO
Catered Lunch

Job summary

Weights & Biases is seeking an experienced software architect to own and evolve its ML workflow systems, including Artifacts, Registry, and Automations. You will design scalable architectures and drive high-volume event throughput across the stack.

The role requires 6+ years of software engineering, strong Go (or compiled language) skills and Python familiarity, plus a track record mentoring engineers and leading initiatives. NY-based team with full-stack responsibilities.

Qualifications

  • Requires 6+ years of software engineering experience with strong proficiency in Go or another compiled language and working knowledge of Python.
  • Candidates must have a track record of leading technical initiatives and mentoring other engineers.

Responsibilities

  • Own the architecture and evolution of core ML workflow systems, including Artifacts, Registry, and Automations.
  • Design and scale systems to handle billions of artifacts and high-volume event throughput across the full stack.

Skills

Go
Python
TypeScript
React
GraphQL
Distributed Systems
Kubernetes
ClickHouse
Kafka
PubSub
System Architecture
API Design
MLOps
Software Engineering
Mentoring
Cloud Infrastructure

Job description

Weights & Biases is seeking an experienced software architect to own and evolve its ML workflow systems, including Artifacts, Registry, and Automations. You will design scalable architectures and drive high-volume event throughput across the stack.

The role requires 6+ years of software engineering, strong Go (or compiled language) skills and Python familiarity, plus a track record mentoring engineers and leading initiatives. NY-based team with full-stack responsibilities.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior ML Workflows Engineer — Scale Pipelines & Mentor
Senior ML Workflows Engineer — Scale Pipelines & Mentor

Weights & Biases • Sunnyvale (CA)

On-site
USD 180,000 - 240,000
Medical insurance
Dental insurance
Vision insurance
+14
Senior ML Workflow Architect
Senior ML Workflow Architect

Weights & Biases • Livingston (NJ)

On-site
USD 180,000 - 230,000
Medical Insurance
Dental Insurance
Vision Insurance
+15
Senior ML Workflow Architect: Scalable Systems (Go/Python)
Senior ML Workflow Architect: Scalable Systems (Go/Python)

Weights & Biases • Seattle (WA)

On-site
USD 180,000 - 240,000
Medical Insurance
Dental Insurance
Vision Insurance
+15
Senior ML Platform Engineer - Scalable Workflows
Senior ML Platform Engineer - Scalable Workflows

Weights & Biases • Bellevue (WA)

On-site
USD 180,000 - 260,000
Medical, dental, and vision insurance
401(k) with company match
Flexible PTO
+2
Senior Software Engineer, ML Workflows - Weights & Biases
Senior Software Engineer, ML Workflows - Weights & Biases

Weights & Biases • Livingston (NJ)

On-site
USD 180,000 - 230,000
Medical Insurance
Dental Insurance
Vision Insurance
+15
Senior Software Engineer, ML Workflows - Weights & Biases
Senior Software Engineer, ML Workflows - Weights & Biases

Weights & Biases • New York (NY)

On-site
USD 180,000 - 240,000
Medical Insurance
Dental Insurance
Vision Insurance
+14
Senior Software Engineer, ML Workflows - Weights & Biases
Senior Software Engineer, ML Workflows - Weights & Biases

Weights & Biases • Seattle (WA)

On-site
USD 180,000 - 240,000
Medical Insurance
Dental Insurance
Vision Insurance
+15
Senior Software Engineer, ML Workflows - Weights & Biases
Senior Software Engineer, ML Workflows - Weights & Biases

Weights & Biases • Sunnyvale (CA)

On-site
USD 180,000 - 240,000
Medical insurance
Dental insurance
Vision insurance
+14
Senior ML Platform Engineer – Workflows
Senior ML Platform Engineer – Workflows

Weights & Biases • San Francisco (CA)

On-site
USD 165,000 - 220,000
Medical Insurance
Dental Insurance
Vision Insurance
+2
Senior Software Engineer, ML Workflows Platform
Senior Software Engineer, ML Workflows Platform

Coreweave • United States

On-site
USD 182,000 - 242,000
Medical, dental, and vision insurance
Company-paid Life Insurance
Voluntary life insurance
+5