Intro0:00
Welcome, everyone. Can you guys hear me okay?
Yeah.
I've been standing here for 15 minutes without saying anything, so maybe we can start now. My name is Abhishek. I'm on the RL and Agent Infrastructure team at OpenAI. What that means is we work on the infra for reinforcement learning, specifically.
And on the product side, we also develop infra that helps run untrusted code as part of ChatGPT, Codex Web, securely and reliably at scale. This talk is called "From Fork to Fleet: Designing an Agent Sandbox Cloud." I'll be very clear that there are a lot of words in the title that have OS and infra concepts, but this is a first principles talk.
So we'll cover what sandboxes are and why they are needed from first principles. We will also try to cover design intuitions around designing an agent sandbox cloud to run sandboxes securely and reliably at scale. So if some of these words don't mean anything, don't be worried.
We'll explain from first principles and go from there.
Last year I gave a talk called "How to Build an AI Sandbox from Scratch." If you're interested, you can look at that talk as well. Think of this as a spiritual sequel to that talk.
Tool Calling1:31
OK, so now let's forget about sandboxes or clouds for a second. Let's just go back in time. ChatGPT came out. It's a very large pre-trained model. People ask all sorts of questions, and it responds really, really well, really, really human-like answers.
But when people ask questions like what is 3 plus 3, or how many R's in strawberry, sometimes it works, and sometimes it doesn't. It answers questions like what is 3 plus 3 quite well because it's trained on the entire internet.
And apparently, people have written 3 plus 3 equal to 6 many, many times on the internet, so it gets itright. But people haven't asked how many R's in strawberry enough on the internet, and so it gets it wrong.
And so it's obvious that for anything code or math-related, or any problem which has a verifiable reward, which means that it can be tested that it's true or false, the model needs something more. And the key unlock was that given the model tool calling capability or a way to execute code, the model gets these verifiable reward questions around code and math correctly.
And thus, from first principles came that if we give the models the ability to execute code, it can hill climb and be very, very good at math, code, and other domains that have verifiable rewards. Like cut to 2026, and we are seeing the consequences of doing that at scale.
So now, how can it answer what is 3 plus 3? And how can it answer how many R's in strawberry? Well, it can write code to do this. So if you see the diagram, we have a training loop.
And the training loop gives it tasks or questions. And then the harness then parses the response of the model. And the response might say, hey, execute code on my behalf. The harness is responsible for executing the code. And then a grader judges whether the answer is correct or not.
And then the training loop backdrops and changes the weights. And that's how we train it to do two things. We train it to call code execution or tools on certain classes or problems. And secondly, we ensure that the code it executes actually solves the problem.
So this is why tool calling is important on the training side. Now let's talk about the product side. In the previous slide, we showed how the models are trained to emit code in order to solve certain tasks. The agent executes the code, and we verify the reward.
Well, all of this is useless if we don't support this model on the product side. So it's basically the same slide as before, but we don't have a training loop. And now the harness is parsing the response and calling the tools and executing the code somewhere,right?
So where is this code or tool being executed,right? This could be your laptop, like with Codex, or any agent you want. Or it could be a node on the cloud with Codex Web or ChatGPT. While we can expect the model to generate non-malicious code, good security practices and hygiene mean that we want to protect our environment, whether it's a laptop or the cloud node.
Attacks can be intentional or unintentional, and we want to protect against those. And it could be like trying to get root on your system or trying to exploit a kernel vulnerability. The models are getting really, really big. And they might try to help you in an overzealous fashion and try to get root to do so.
So you want to try and avoid and make sure it doesn't attack the node or where it's running. Thus, this is where the sandbox comes in. We need the sandbox to run this untrusted code and ensure that it can do its work, but it shouldn't be able to exploit any vulnerabilities and get root on your system.
In the cloud, other sandboxes with other users' data might be running as well. And we don't want it attacking and getting data of other users out. So a sandbox is an environment in which you can run these tool calls and execute code on behalf of the model securely.
And it could be on your laptop, or it could be on the cloud.
I think that everyone's probably used OpenClaw or any agent like that. I think it's a very, very, very big peek into what's to come. A lot of these agents are running locally on your laptops. And I think it's kind of a slap on the face for 20 years of cloud computing that everyone's running this locally on their laptops.
And if you see the image, I don't know if this is an actual product, but it's very funny because the lid of the laptop is open. It's because you don't want your agents to sleep. Well, we have a whole cloud.
There are industries built on this thing. So I think the future is us running your agents in the cloud. They're persistent, long-running. And I really, really hope that moving forward, this is not something you see anywhere. But people rented VPSes.
They ran OpenClaws on Hexner or Mac Minis in the cloud. So I think OpenClaw was a very, very good peek into what might be for sandbox clouds in the future.
We've discussed why sandboxes are important in both research and product, but they have slightly different needs. On research, we want to optimize for throughput. We want to run many, many training loops at scale and have many, many rollouts.
Design Pillars6:45
A rollout is one version of a task. So what is 3 plus 3? And you might have five answers to it, and one answer is a rollout. We want to take many, many shots on goal in parallel. And so throughput is very, very important in research.
In product, latency is very important. I think any successful product in the last 20 years has been super fast. So if you don't start a sandbox in time and you don't execute code fast enough, people will churn from your product.
Reliability is important on both sides. If you fail constantly, you've wasted GPU tokens on both sides. And GPU is goldright now. So you want to just make sure the tokens you're getting are useful. And similarly, if your agents aren't reliable on the product side, it's game over,right?
People will churn from your product. Security is also important. On the research side, we are training on OpenAI infrastructure. So if a model gets root and it's not aligned, it can try to attack OpenAI infrastructure. It can exfiltrate and release our model weights or whatever.
And then on the product side, it can exfiltrate other users' data, attack the infrastructure as well. So security is important for both research and product. So today, we'll focus on these three pillars. There are many, many parts of a sandbox cloud, but we'll specifically focus on runtime.
So how can we run a sandbox on one node securely? Secondly, we'll focus on persistence. I think compute was the first unlock. People realized you give sandboxes a Linux computer, and they do crazy things because they're pre-trained on so much Linux data.
But I think now, if you help them, if you give them a computer with an actual disk that can be saved, then they become a true knowledge worker. And the last part is orchestration and how to run these at scale for many, many users at ChatGPT and Codex scale.
So before we start, this is a first principles talk. So let's discuss how Linux executes code on your machine,right? So on Linux, a thread is the smallest unit of execution. The kernel is the thing that provides privileged access to a thread via something called system calls or IOCTELs.
Linux Basics9:06
So
whenever the user space program wants to access some hardware or some privileged resource, it needs to talk to the kernel and call this operation that switches the hardware context to a more privileged context. And so let's look at what that looks like.
So if you see here, your processor has different rings of execution. So based on which ring you're executing in, you get different privileges. So the kernel mode is executing in ring 0. It has the highest privilege. And anything in user mode is running in ring 3.
Whenever we want to access privileged resources, we call system call instructions that change the context of the CPU. And so there are two attack vectors in a Linux system,right? First is getting root. In getting root, you're still in ring 3, but you're the highest privileged user on the system.
So you can actually pretty much do anything on the system. You can read your SSH keys, encryption data, et cetera. And then the second version is actually running a kernel mode exploit. So running code in ring 0, this is terrible.
You can actually dump processes' memories. And I don't even want to say what all can happen. So you can get root, and you can have kernel exploits. And if you get kernel exploit, it's like a New York Times article waiting to happen.
So these are the two attack vectors on a Linux system. So with that background, let's design the simplest way to execute tools on a Linux system. For example, we can literally have an API server that your harness is calling.
And for every tool call, it can fork a process, exec the tool that the model needs, and just have a fork and exec model,right? So now there are a couple of problems with this model,right? A, as we discussed the execution model, the forked process can now directly talk to the kernel.
And so the model can try to attack the kernel, get root, or try to get a kernel exploit,right? The second thing is, imagine you have a while loop and just forking processes in one tool call,right? So now you've kind of become a bad neighbor or a noisy neighbor.
And you've kind of brought down the node, and no other tool calls can run,right? So fork exec is the simplest thing you can do. It has one thing going for it. It's the most performant solution because it's as fast as just forking something.
It's native performance. But everything else is very bad about this solution.
So we saw that forking exec was bad. There was a noisy neighbor issue, and there's a security issue. We now turn to something called containers on Linux. You must have heard of Docker and other things associated with containers.
Containers12:04
But remember, this is the first principles talk. So let's go deep into what containers are in their very raw form. Containers rely on two concepts on Linux, namely namespaces and C groups. Namespaces are for resource isolation, and C groups are for controlling the amount of resources a set of processes can consume,right?
So let's see in depth how they can help our problems. So in this diagram, you can see that there are different types of namespaces a set of processes can have. So on the left, you can see this container has a PID namespace.
So inside that process namespace, it looks like we have a process hierarchy of PIDs 1, 2, and 3. But if you look at the PIDs from outside the container, they're just regular processes with other PIDs,right? So you've abstracted one resource, which is the process inside this container.
Another example is a mount namespace. So if you see the bottom half of the diagram, you can mount any file system on top of any mount inside the container. But from outside, the original mount is still visible,right? So this is a nice way of isolating different resources in a container.
And there are many namespaces like PID, network, mount, et cetera.
So moving on to the second principle of containers,right, like C groups. So now we have a way to isolate resources within containers via namespaces. But remember, I gave the example of someone doing a while with fork,right? How can we control that?
Well, we can have C groups and control how much CPU and memory a container can consume. And that way, we can make sure that one container doesn't bring down the entire node. So namespaces and C groups provide some amount of isolation and control on the system resources to not bring it down in case it's malicious or bad.
If you've seen the previous diagrams, the fundamental problem with containers is that there are still native processes running on the host. So even if we abstract resources, two containers can't attack each other because their resources are isolated. But a process in a container can still exploit the kernel boundary and try to get root or a kernel exploit,right?
And once they can get root, they can attack other people's data from other sandboxes, exfiltrate data, and try to get control of the node, like we discussed before.
So security is a spectrum. You can still get a decent amount of protection from containers. One example to do this is something called seccomps. So you can actually have a filter on the amount of system calls your container can call.
And you can also control the arguments each system call can take. So you can say that you're reducing the attack surface of the kernel for this container and kind of have some amount of sanity on how it can attack the kernel.
The problem with this is that many times, you don't know beforehand what system calls some container might call,right? So now you're blocking requests for users. And later on, you have to change the seccomp filter to allow something or not allow something.
So the feedback loop is pretty bad for a product like OpenClaw or any agent that wants to do crazy things,right? You want to make magical experiences happen in the sandbox, and you don't want to restrict them,right?
So yeah, this is the fight. To sum it up, containers interact with the same host kernel. So they do have some protections, but at the end, it's the same host kernel they're trying to attack,right? And they can get root and kernel exploits, et cetera.
So we went from forking to containers. Raw fork was obviously very, very bad for all the reasons. Containers are better than fork because they still have some protection, and you can still have seccomp for reducing the attack surface.
Can we do better? Can we reduce the host kernel exposure a little bit more?
Yeah, so the second thing we can discuss is to solve the attack surface problem is something called gvisor. The crux of the security boundary is the kernel API from before. So gvisor uses this point as its key security story.
gVisor16:27
So gvisor implements a lot of syscalls in user space. You can think of it as an application kernel. Its sentry is a user space kernel written in Go. And the file system is accessed by another daemon called the gopher.
So the sentry implements the Linux API itself, including process management, workload management. And whenever you call a system call, it's intercepted and it's serviced by this user space program. So then you can argue that you're not implementing and attacking code that's running in ring 0 or kernel mode.
You're actually running in user space and ring 3. And any exploit you have is still in user space. So it's better than you exploiting the kernel directly. But the fundamental problem is that the sentry and the gopher, just going back, the sentry and the gopher are still on top of the host kernel.
So if there is an exploit on them, you can still have a chained two-step exploit. So you first exploit a problem in the sentry or the gopher, and then you exploit from the gopher to the kernel,right? You can still get to the host kernel eventually.
And with these models of 5.6 and other types, you can have them figuring out bug reports and these things and trying to chain exploits,right? So you can still get to the host kernel,right? So yeah, the chain is harder here because it's a two-step chain compared to the others, but it's still reachable.
Can we do better than this?
Yeah, that's the same slide. It's saying that in all of these solutions, you can eventually get to the host kernel, whether it's fork, gvisor, or containers.
So the key question is, can we have a way in which untrusted or malicious code can have exploits, but they can never or find it very, very hard to exploit the host,right? So even if I can get root, even if I can exploit the kernel, I still want my host to be protected,right?
MicroVMs18:30
That's the final goal we have. So do we have something that we can use for this? So if you've known some of my background, you'll know where I'm going with this. So it turns out Linux provides a very, very nice thing called virtualization, which is hardware-powered at the CPU level.
And it provides an abstraction at the hardware level, which means that even if you can get root or execute in ring 0 in the guest, your host is still protected. And this happens because the guest kernel runs in ring 0, but in a separate processor context called VMX non-root, while the host kernel and the hypervisor run in ring 0 in something called VMX root mode.
So ring 0 gives the guest kernel full control inside the guest, but no control on the host. So you can exploit the guest all you want, but the host is still protected. The processor has to switch between the guest and the host whenever the guest wants to access some privileged resource.
We'll see how that works. But that is a key trade-off here. There's a performance penalty you pay every time the CPU is switching back and forth between these two modes. Let's see with a diagram what I mean. So you can see in the old diagram, we showed ring 0 and ring 3 for executing user space and kernel code.
But here, the guest kernel and the guest user space are running in separate CPU contexts. So this is guaranteed at a hardware level. And so even if you can get to ring 0 in the guest, your host is protected from it.
Yeah, now let's talk about what does para-virtualize and hardware-based virtualization mean. This is the key of all the VM sandboxes you might have seen on Hacker News or Reddit. So we'll go from step by step,right? So on theright-hand side, you see something called the VMM.
Can everyone see that on theright-hand side? The VMM stands for Virtual Machine Monitor.
This software is called a Virtual Machine Monitor. And you might have heard of QEMU when you search for Linux virtualization. QEMU and other VMMs, their sole job is to talk to DevKVM. So you see that arrow going down.
The DevKVM is the hypervisor API of the Linux kernel. All QEMU and other VMMs do is set up the kernel, root FS, allocate some memory for this guest virtual machine, and call into the DevKVM API.
As discussed inside, when the guest runs on the left-hand side, it's running as a completely different Linux system. It doesn't know what's happening on the host side. But it has these devices, like a block device or a net device, coming up as regular Linux devices.
But when it actually tries to access the devices, they exit out to the host context. And if you see the block backend and the network backend, there's actually code running, emulating the devices in the host. So whenever the guest exits out, it's being serviced by the VMM process and these device processes.
So what do we mean by para-virtualization? So para-virtualization means that we want good performance while we want to access hardware in the guest. So the guest drivers, when the user space talks to them, the guest drivers are aware that they're running in a virtual machine.
And so they talk via something called Vertio, which is a more efficient way of the guest and the host talking to each other. And so inside, there are just PCI devices like any other hardware. And when you talk to the PCI devices, you magically exit out onto the host.
So 15, 20 years ago, a bunch of Linux wizards made this thing happen and made this performant. And it truly is magical, how it works reliably and performantly across different hardware processors. And so from the host's point of view, when this is running, you just see a block thread.
If you do a PS, you'll see a block thread, but that's running on the guest context. And when the thread exits out, the thread finally wakes up. It wakes up in this VMM process, and the process runs, and it's running inside.
Yeah, so we had this seismic shift in 2023 where a bunch of new VMMs came. For a long time, it was just QEMU that was used for running virtual machines. But QEMU has a lot of craft. It supports many, many architectures.
It has many devices. And it's written in C. And historically, many, many escape attacks were attacking the devices written in C. So the first Rust-based VMM was something called CrossVM, which our team at Google wrote when I was there to support Linux virtual machines on top of Chromebooks.
And the key point here was that we don't need all the craft of QEMU, and we can use Rust to be a memory-safe implementation of this tricky system software. And secondly, we have these emulated devices that we can jail.
So if you attack the block device, we've only given it permission to access block resources. So you can't access network resources. Similarly, if you attack the net device, we've jailed it, and we don't give it access to block resources.
So then you can still have a second gate of security. So both Rust-based safety, like being memory-safe, and just jailing at a more granular level at the device provides better safety than just QEMU.
So you must have heard this word microVMs everywhere. And no one really answers why the word micro comes. So it turns out it has nothing to do with what's running inside the guest. It's everything to do with the VMM itself.
So all these new-age Rust-based VMMs, they have a much smaller memory footprint because they don't support as many devices. And they also boot much faster because they don't have as much craft. So a mixture of just less bloat and just booting up faster is why the industry has called them microVMs.
One second. Yeah, sorry. And then you must have seen Firecracker and Cloud Hypervisor. There's a lot of confusion around who came before. CrossVM was the first Rust-based VMM that came before. And then Firecracker forked CrossVM, and it's used on Amazon for their Lambda and serverless load.
And Cloud Hypervisor is a more general VMM that many, many companies contribute to. And when you historically see microVMs on the internet, it's powered by one of these VMMs, basically.
So it may seem complicated, but at the end, everything is APIs. MicroVMs are no different. So here's a small view of how you can actually start a microVM. So your harness actually forks a Cloud Hypervisor binary process. When the process starts, it exposes an API over a Unix domain socket.
And if you can see in step two, we call the create API, give it the root FS, kernel, CPU, and memory. And then we finally call start. And when we call start, you can see from previous diagrams that the VMM literally calls into DevKVM, just like before.
And it starts these guest microVM.
And then once the guest is running, in this case, the agent sandbox is running, it might be that I want to talk to something inside the sandbox to say that, hey, save your state, or I'm attaching some devices, or I'm doing some XYZ operation.
So generally, in agent sandboxes, you have a PID1 that exposes an API server. And your harness or something on the outside is talking to this API server. And then this is the way how you can use microVMs to be agent sandboxes that can run more securely than the other primitives that we showed.
In this case, we are using VSOC, which is a socket that can help you communicate between the guest and the host. Or you can use the IP stack on the node itself.
And so there are no free lunches in systems. And so there are some trade-offs. So obviously, you get really, really good isolation using microVMs at a hardware level. You can still attack the host, but it's much harder. You have to attack the KVM stack, and then you have to attack the device.
The chain is much, much harder to do. But it has been seen, and I'm sure it will be seen more with these models. You can jail the devices granularly. So you can use SecComp and other security hardening things for the devices.
So even if you get compromised on one device, your whole system cannot be brought down. Like I said, there is a performance overhead. When you are exiting and entering the host and the guest context, it's a very, very heavy operation, and you pay in performance.
Memory sharing is not as easy. There's something called a balloon driver. And so you have to actually ask the guest to give back memory and reclaim it. So it's always a reactive thing. You can't immediately reclaim and claim memory.
And then a lot of sandboxes these days have GPU access, presumably for auto research or some sort of ML research type of agents. This is not as easy with microVMs. There's something called Vertio GPU, but that provides high-level graphic library type access.
For direct metal access, there's something called VFIO. But it can only be shared by one sandbox at a time. It cannot have multi-tenant things. But given all of this, my view is that system tricks can cover performance issues, but they cannot hide security breaches.
And as a company, you can lose trust once, and it's very hard to regain. So I always prefer the more secure solution and try to make up with system tricks for performance issues. And in my history of working on sandboxes, I've seen there are, I would call it, the seven stages of grief, the seven stages of sandboxing.
In the end, everyone always wants a VM because they tried everything. They tried containers, gvisor, v8. And then they realize, oh, I want a whole Linux box because I don't want to end up without XYZ functionality. And if I want a whole Linux box, I want to obviously be secure.
So if you're a startup or a founder in this space, let me save you the story and two years of grief. Just please use microVMs from the start. And then if it doesn't work, tell me, and then we can talk about other things.
So now we've discussed how to run untrusted code on one node. And I think now people have woken up to the fact that these models are very, very good drivers of Linux boxes. So if you give them a computer, they can just pretty much do magical things, as we've seen with OpenClaw.
Persistence29:45
They just pre-train on a lot of Linux,right? However, imagine if I gave you a computer without a disk. Every time you close the laptop, your data and your work goes away,right? That's not a fun world to live in.
And somehow, the agents in the cloud at least are in this sort of worldright now,right? So we want to give them durable storage. And so this part of the presentation is specifically working on disk storage, not memory persistence, but disk persistence.
So let's see why it's important and how we can do it.
So as shown in previous diagrams, your microVMs have disks attached to them. There might be just files or other disk devices on the node that you pass through to the VM. And so from a product perspective, the tasks that the users are doing in these sandboxes are becoming much more complicated and much more longer horizon.
So people are making presentations, and entire GitHub repos are being created inside the sandbox. Now imagine if the node dies on the cloud or the model has a flake, and you created this presentation, and you just lost it,right?
It's A, bad for us because we wasted a bunch of GPU tokens. It's obviously bad for the user because you did a lot of this work, and you lost it. So not just from a product perspective, but just from a good experience and utilization perspective, we need to have some way to save the disk state of the sandbox.
And let's go into three big use cases on what persistence can unlock,right?
So counterintuitively, persistence actually helps reliability and scale. They might seem likeorthogonal concepts, but they're very much related. So for instance, you have a long-running task in a sandbox that has many packages installed, and you've created GitHub repos and presentations and things,right?
If you keep checkpointing it periodically, and if the node fails or the cluster fails, you can now restore the sandbox in the exact checkpoint state on another node. You can also do it intentionally. If you want to upgrade a cluster or do some A/B testing on nodes, the persistence has now let you reliably run sandboxes across your Fleet,right?
So it's a very good way to scale and be reliable.
Like I said, how many of you have used Gold Mode in Codex or know what it is? Yeah, amazing,right? So if you run Gold Mode, now it's like, I think, three days is my record for running something. But now people are doing longer and longer tasks.
And I think this trend will continue in the cloud as well. So to support this, we obviously need to have checkpointing so that the model can save state, restore it on another node, and you can keep going forward and forward,right?
And so you're resilient to any failures across in the infrastructure. This is the most interesting part, actually,right? So if your harness wants to explore multiple solutions or sample spaces, it can actually checkpoint the sandbox state. And it can do a Monte Carlo research and go ahead and backtrack, checkpoint again.
So this way, it can actually do rollouts over many, many days and come back with the actual solution,right? And I think my sincere hope is that if we make this infrastructure correct, we can help solve diseases, find new drugs, because the model can just keep going on for longer and longer,right?
But it doesn't happen till we have really, really rock-solid primitives to do this.
And so given the needs and the
pillars we want to support, here are some of the things a disk snapshotting solution should support,right? First, at ChatGPT or Codex scale, we want to do incremental snapshotting. And what that means is, if you call snapshot twice, I'm just snapshotting the diff between the two snapshots.
Otherwise, if I have to save gigabytes of data at every turn, I'm going to bankrupt the company. And it's just a slow experience regardless,right? The snapshotting API itself should be very, very cheap and fast so the model and the harness can keep snapshotting and exploring very fast.
And similarly, just like creation should be fast for products, restoring is just nothing but just recreation from a snapshot. And so restoring should also be very, very fast for a good product experience. And then we have two paradigms we'll discuss.
One is always saving, in which the harness doesn't have to explicitly call a save API, versus explicit saving, where the harness is calling save, save, save. We'll see how we can implement both. So I'll give a reference solution.
And then there are some more design choices with disk snapshotting,right? So first, like I said, on the left-hand side, do you want incremental or full snapshots? I argue, I think at our scale, we want incremental snapshotting. Secondly, do you want to snapshot the entire root FS, or do you want to have certain folders like workspace or mount something, something that you want to keep that configurable?
And lastly, we can see how you can snapshot at a file system level or at a block. So in Linux, disks are nothing but block devices, and each file maps to different blocks. We'll see after this. And so you can do very, very efficient snapshotting by just zipping up the blocks that have changed for a file.
Or you can do entire files that have changed,right, for more write amplification. And the high-level flow of snapshotting is basically, you figure out what's changed, you zip it up, put it to the cloud, and then when you restore, you pull it down and you restore the microVM,right?
And so this is a first principles talk. And so I just wanted to go over how Linux storage works from first principles. So Linux represents disks as block devices. So you see on theright, it thinks of a block device as having logical blocks from 0 to n.
Snapshotting36:00
A file system maps directories and files into something called an inode data structure. And the inode says, OK, offset 0 in file F maps to this block, logical block, in disk D. And then on the disk itself, there's firmware running, which says, oh, logical block 1 is like sector 100 or sector 500 or page XYZ.
So the hierarchy is on these logical blocks up to the file system. And we leverage that for block-based snapshotting. And then within a microVM, there are two ways of accessing storage. On the left-hand side, you can think of this as sharing a folder like Google Drive.
But it's very, very inefficient because the VM is doing file system operations, exiting at every file system operation, which is very inefficient. On theright-hand side, you actually give a disk-like abstraction at a block. So you give a block device to the microVM.
And this is way more efficient because you can use the caches inside the guest, and you don't have to exit out as much. You only exit out when you truly need to access the block device, and the host has to service you.
This is the high-level diagram of how you will have always-on persistence in disk snapshotting. So if you have two microVMs, they have a block device. They will see this dot image as a block device inside. And as they're writing to the block device, we are writing through to the cloud.
So there's some sort of distributed file system that we mount inside that's giving this always-on persistence. And like I mentioned, for explicit persistence, we have an actual API called the save API. Calling the save API from the harness figures out what's changed between the last snapshot.
It bundles up this diff into an artifact and returns a snapshot ID. Later on, you can give us this snapshot ID. We figure out the lineage of snapshots that make this snapshot ID. And then we download them one by one and apply it on the node, and you get a microVM restored with this thing.
So let's see how we can implement this,right?
Give me one second. I want to see how we are doing on time.
I think we have five minutes left, so we'll go fast here. Yeah, so we can do explicit persistence using something called copy on write. So Linux has these XFS file systems, XFS-like file systems. So you can do pretty much have zero latency copies because you don't change any blocks when you copy.
But when you actually change the blocks on a file, that's when you pay the penalty. And so in this design, we have a base dot image, which might be the Codex base image or whatever, like the ChatGPT base image.
We create a zero copy on top of it, which is the writable layer. And then when you write to it, now you're changing the blocks in this layer. And then when you want to snapshot, I use something called FIE map, which tells me what blocks have changed and what ranges.
I zip that up and store it to the cloud. And I can do something very nifty here. I can actually lie to you while I'm uploading to the cloud. So the snapshot can happen, return very fast as I'm uploading in the background.
I don't have to wait till I'm uploading. And then on the other side, I have this diff. I download this artifact. I figure out what extents have changed. And then I apply it back on top of the base image, and I start the microVM again.
So now we have a restored sandbox with the exact same state at a block level,right? Now, how can we do always-on persistence,right? So this is one way of doing this. I know there's NFS also and other distributed file systems.
But NFS, for instance, isn't as performant, and it's not POSIX compliant. And I think our models are just very good at anything POSIX compliant and standard. So you can write a file system actually on top of a GCS or S3 or durable block storage.
And you can use something called NBD. So within the sandbox, you'll actually see a block device. But inside, it will have literally a tiered cache of blocks that are persisted first to an in-cluster cache. And then the in-cluster cache is writing back to the block storage, the object storage.
So you have this nice global tiered architecture where you're actually caching things at the block level and finally inside the microVM and get a very performant file system inside.
Yeah, so that's the persistence part of the presentation. And so the one takeaway I want you guys to think about is, I think storage is the next unlock here. As you're working on sandboxes, think of what all you can snapshot and restore fast to give this new paradigm to harnesses so they can recover from failures and explore, do Monte Carlo-like searches.
So we've discussed running things on one node. And of course, if our one node dies, we are done. So we want to be able to run across many, many nodes across the world. So I'm not going to mention about Kubernetes or other orchestration things.
Orchestration41:22
This is a first principles talk. So we'll lightly hint about the challenges here. So ideally, we want multiple machines to support the runtime we discussed. So we can group nodes into clusters and spread the clusters across the regions.
A top-level control plane chooses a cluster using region, load, and other factors. And it's not very different from the orchestrators you're familiar with. These ideally choose a cluster close to your ChatGPT cluster so you can have fast access to the harness.
Inside the cluster, there's a scheduler also. And the scheduler tells you which node to pick based on the load and other factors,right? So if nodes are dying or failing, it won't choose that. It will intelligently route the sandbox to the thing.
And again, low latency and reliability remain very, very key north stars for this architecture.
And so here, we can use some microVM features to support low latency creation ideas. So a lot of systems cheat for low latency. They pre-warm sandboxes, and they pick one, which is great. Another way to do this is you can actually take a memory snapshot of a microVM and just in time start it in milliseconds as the request comes.
And so you can leverage this nice microVM property that you can save the guest memory and start from that. And the third one is a hybrid solution. So you can have a warm pool. But as it's growing, you can grow it from the memory snapshot.
So you can get best of both worlds. The trade-off for a warm pool is that you're consuming CPU and memory in idle state. Ideally, you don't want to do that,right? So there's a trade-off between 1 and 3 and 2, basically.
And so here is a way where we can use snapshot restore for better orchestration. So remember, we discussed that a snapshot can have a lineage of many, many layers. So once you want to restore from a snapshot and you find out you have, let's say, four layers that you want to pull down, and that makes the lineage, you can actually smartly route you to a node which has to download the least amount of stuff.
So in this diagram, you can see node A has some layers, node C has some layers, but node B has all the layers that you need. So the scheduler then routes you, and it gives it the highest score and routes it because it has all the snapshot layers.
So you can use snapshot with orchestration to just have faster creates and even just more reliable orchestration. Yeah, this is the talk. And hopefully, it gives you some design intuition around sandboxes, why they're important, what's the next unlock.
And I want to see more of you guys using it in a secure way. Thank you.





