Software Engineer, Baseten
Storage Systems, Agent Infra
$160-330k
+ Equity
ML application builder for data scientists
Be an early applicant
ML application builder for data scientists
101-200 employees
Be an early applicant
$160-330k
+ Equity
101-200 employees
Company mission
To make machine learning accessible to every organisation.
Role
Who you are
- Strong systems programming background in Go, Rust, and/or C/C++, with comfort reading kernel code
- Deep understanding of Linux filesystem and storage internals: VFS, page cache, dentry and inode caches, FUSE, the block I/O stack, fsync semantics, and crash consistency
- Hands-on experience building distributed storage systems, content-addressable storage, or large-scale caching infrastructure
- Experience profiling I/O, memory, and RPC bottlenecks and optimizing tail latency in production
- Ability to debug correctness bugs that only show up under concurrency or partial failure
- You move fast and like owning a problem end to end on a small team
Desirable
- Experience with a distributed filesystem or object store such as JuiceFS, CephFS, or S3-compatible storage, or contributions to those projects
- Knowledge of consensus and replication protocols (Raft, leader leases, quorum reads)
- Experience with snapshot, clone, and copy-on-write filesystem features
- Experience operating multi-tenant storage with hard isolation requirements
- Familiarity with Linux performance tooling such as eBPF, io_uring, perf, ftrace, and blktrace
What the job involves
- Storage decides how fast agent sandboxes start and how safely they share data. Agents need a filesystem that thousands of sandboxes mount at the same time, with strict isolation and the consistency agents need. Agent context is mostly small files, so the access patterns look very different from storage built for large files
- The Agent Infra team builds Agent Drive, the distributed filesystem that agent sandboxes mount over FUSE, and the storage underneath it. The work uses tiered caching, replication, content-addressed data, and Linux filesystem internals. In September 2026, Baseten acquired Blaxel, whose sandboxes and Agent Drive filesystem are the foundation for Baseten Sandboxes
- We are building a cloud, not just using one, so storage is designed for many tenants from day one. The team is small and ownership is big: we ship in weeks and months, not multi-year programs
- If you care about page cache coherency, FUSE, replication protocols, and content-addressable storage, we'd love to talk
- Design and implement core Agent Drive features: metadata replication, volume placement, garbage collection, and FUSE client behavior
- Solve multi-attach correctness: cache coherency across many FUSE mounts of the same drive, dentry and inode invalidation, write ordering, and crash consistency
- Optimize tail latency on read and write paths, including page cache strategy, writeback, prefetch, and metadata RPC batching
- Build tenant isolation into the storage layer, including quota enforcement, blast-radius containment, and noisy-neighbor mitigation
- Own the storage side of sandbox startup and resume performance
- Build durable block storage for sandboxes alongside other Baseten storage teams
- Build observability into every layer, drive storage incidents end to end, and turn each one into a permanent fix
- Contribute upstream to the open-source storage projects we build on
Our take
As with its mathematical namesake, Baseten focuses on simplifying the complex. The San Francisco startup focuses on simplifying the integration of artificial intelligence into businesses. The tools that Baseten build demolish the need for businesses to hire specialised engineers, utilising the huge strides in artificial intelligence that we’ve seen in the last few years, all without adding workload to businesses.
With serious funding behind it, Baseten has grown its team by three times year-over-year over the last five years. Specifically, it's hired across engineering and marketing, gaining talent from companies such as GitHub, Google, Uber, Amazon, and Palantir. Part of Baseten's focus is on helping companies implement the tools that they have at their disposal, a long-overlooked aspect of machine learning that many businesses have been eager to champion, but are uncertain about how to actually use.
Simplifying the complex world of artificial intelligence for business is no mean feat, but with hundreds of millions in funding behind it and a $5 billion valuation reported in 2026, the company is well on its way to market domination. What's more, the three founders, who met whilst working as engineers at Gumroad, have a lot of experience in navigating this gap in the market.
Kirsty
Company Specialist at Welcome to the Jungle
Company
Funding (last 2 of 7 rounds)
Jun 2026
$1.5bn
SERIES F
Jan 2026
$300m
SERIES E
Total funding: $2.1bn
Company benefits
- Remote-first work environment. The Baseten team is welcome to work from wherever they want; fully remote, in our San Francisco office, or a mix of both. Today, our team (including our founding team) is spread across the United States, Canada, and Armenia. We provide a $1,000 stipend for you to make your home-office comfortable and productive
- Regular in-person team summits. We get together as a team three times a year to plan, workshop, and most importantly, get to know each other better
- Unlimited PTO. We ask that everyone take at least 4 weeks of vacation. And we have a company-wide break between Christmas and New Year's Day
- Full healthcare coverage. Medical, dental and vision insurance for you and your family
- Paid parental leave. 16-weeks fully paid parental leave (adoptive and non-birth parents included) and flexibility with schedules while returning to work
- Company-sponsored 401(k) for you to contribute to
- Learning and development budget. We encourage you to take classes, attend conferences, and invest in your craft and we’ll cover expenses to make it happen
Company HQ
Union Square, San Francisco, CA
Leadership
Tuhin Srivastava
(Co-Founder)Previously Co-Founder at Shape, Machine Learning at Gumroad and Skulpt.
Amir Haghighat
(Co-Founder)Previously Head of Engineering at Gumroad, Software Engineer at Yelp and Engineering Manager at Clover Health.
Phillip Howes
(Co-Founder)Previously Co-Founder at Shape, Machine Learning Engineer at Gumroad and Research Assistant at the University of Sydney.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 82 more jobs at Baseten