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

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
check
Comp. & Benefits
badNo H1BnoteU.S. Citizen Onlynote

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

Software engineering experienceComputer systems expertiseContainer orchestrationBuilding fault-tolerant systemsProficiency with PythonGPU resource allocationEmpathy for practitionersCommunication 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
Daniel Gross,Nat FriedmanNVIDIABond,Felicis
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