Weights & Biases · 4 hours ago
Staff Software Engineer, Infrastructure- Weights & Biases
Weights & Biases, acquired by CoreWeave, is at the forefront of developing an end-to-end platform for AI development and deployment. As a staff engineer on the marimo team, you will be responsible for designing and implementing backend architecture that ensures high availability and low latency for their cloud-hosted notebook service.
AI InfrastructureArtificial Intelligence (AI)Data VisualizationDeveloper ToolsGenerative AIMachine Learning
Responsibilities
Co-design and implement the backend architecture of molab
Solving for high availability
Low latency
Stability
Fraud and abuse
Design molab to run on CoreWeave's specialized kubernetes-based clusters
Integrate with CoreWeave object storage
Solve for keeping utilization of GPUs high
Qualification
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
Weights & Biases is a developer-first MLOps platform that builds machine learning performance visualization tools.
Funding
Current Stage
Growth StageTotal Funding
$250MKey Investors
Daniel Gross,Nat FriedmanNVIDIABond,Felicis
2025-03-04Acquired
2023-09-01Secondary Market
2023-08-09Series Unknown· $50M
Recent News
2026-02-06
trinityventures.com
2026-02-05
Qualcomm Ventures
2026-01-20
Company data provided by crunchbase