> Markdown version of [/jobs/ext/4618349-staff-software-engineer-infrastructure-marimo](https://www.wearedevelopers.com/jobs/ext/4618349-staff-software-engineer-infrastructure-marimo). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Staff Software Engineer, Infrastructure- Marimo

- **Company:** Weights & Biases
- **Location:** San Francisco, CA, United States (Remote available)
- **Experience:** Expert
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Cloud Computing, Information Engineering, File Systems, Distributed Systems, Fault Tolerance, Inter-Process Communication, Python (Programming Language), Machine Learning, Multiprocessing, Natural Language Processing, Software Engineering, Graphics Processing Unit (GPU), System Availability, Concurrency, Parallel Computation, Backend, Containerization, Kubernetes, Information Technology, Low Latency, Front End Software Development, Asynchronous Programming
- **Published:** November 7, 2026
- **Apply:** https://www.juju.com/job/28_25303_df3349a60852d3406eac2f481d627fa1

## About the Role

* 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

Wondering if you're a good fit?

 * You love to build empower developers by building performant and scalable tools
 * You're curious about how developers in AI, ML, and data experiment with models and code
 * You're an expert in building highly scalable cloud infrastructure

## Description

As a staff engineer on Marimo's molab team, you will co-design and implement the backend architecture of molab, solving for high availability, low latency (both the ability to rapidly spin up and spin down notebook kernels on demand, as well as low latency communication between the notebook frontend and backend), stability, and fraud and abuse. You will design molab to run on CoreWeave's specialized Kubernetes-based clusters and integrate with CoreWeave object storage, and will solve for keeping utilization of GPUs high.