Principal Software Engineer, Infrastructure - Weights & Biases jobs in United States
cer-icon
Apply on Employer Site
company-logo

Weights & Biases · 3 days ago

Principal Software Engineer, Infrastructure - Weights & Biases

Weights & Biases, acquired by CoreWeave, is focused on empowering developers with tools and infrastructure for AI. The Principal Software Engineer will design and implement the backend architecture for the molab team, ensuring high availability and low latency systems.

Artificial Intelligence (AI)Data VisualizationDeveloper ToolsGenerative AIMachine Learning
check
Comp. & Benefits
badNo H1BnoteU.S. Citizen Onlynote

Responsibilities

Design and implement the backend architecture of molab, solving for high availability, low latency, stability, and fraud and abuse
Design molab to run on CoreWeave's specialized kubernetes-based clusters and integrate with CoreWeave object storage
Solve for keeping utilization of GPUs high

Qualification

Software engineering experienceComputer systems expertiseContainer orchestrationBuilding fault-tolerant systemsProficiency with PythonAI/ML stackGPU resource allocationEmpathy for practitionersCuriosity about developer toolsCommunication skills

Required

8+ years of experience in software engineering
Strong fundamentals that are language agnostic
Expertise in computer systems, including parallel computing (threading, multiprocessing), concurrency (asynchronous programming), networking/inter-process communication
Experience with containerization, container orchestration (kubernetes), scheduling, networked filesystems, resource allocation, distributed systems, and cloud infrastructure
Experience building highly available, fault-tolerant systems
Strong communication skills, written and verbal

Preferred

Proficiency with Python and Python packaging
Basic experience with or awareness of the Python stack for AI/ML
Empathy for practitioners and researchers in AI, ML, data engineering, NLP, or other quantitative work
Experience with GPU resource allocation and sharing

Benefits

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 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

Weights & Biases

company-logo
Weights & Biases is a developer-first MLOps platform that builds machine learning performance visualization tools.

Funding

Current Stage
Growth Stage
Total Funding
$250M
Key Investors
NVIDIAInsight PartnersCoatue
2025-03-04Acquired
2023-09-01Secondary Market
2023-08-09Series Unknown· $50M

Leadership Team

leader-logo
Chris Van Pelt
Co-Founder & CISO
linkedin
leader-logo
Shawn Lewis
Founder/CTO
linkedin
Company data provided by crunchbase