[
  {
    "startMs": 12440,
    "text": "Go."
  },
  {
    "startMs": 13020,
    "text": "Okay. Hi everyone. Welcome, um, this 20-minute talk about connecting AI to loads of legal documents. My name's Jacob. I'm an engineer at Legora."
  },
  {
    "startMs": 26260,
    "text": "Yeah, and I gotta step into frame here. Uh, I'm Simon. I'm the CEO and, uh, CEO and co-founder of TurboPuffer, a search engine that we work with Legora and others on."
  },
  {
    "startMs": 35840,
    "text": "It, um, super quickly, um, introduction to Legora. We're a collaborative AI platform for legal work, and so that means we have law firms that are clients, and we have, uh, in-house legal teams that are clients, and they use Legora to do reviews of contracts. They use it to go through an absurd amount of contracts and make sure that they all look good. They use them to, uh, create new contracts. They do legal research, which means, um, looking over all, uh, potential law,"
  },
  {
    "startMs": 66160,
    "text": "and they collaborate inside Legora. So you can think of Legora sort of as a Linear/Figma/Notion/GitHub for legal work. It's a lot. We, um, are one of the fastest growing companies, uh, right now. We grow extremely fast. Yeah, tons of numbers on the screen. I'll just skip through that. What we really wanna talk about is search today. So at Legora, there's two types of search that we do. There is project search and legal research.  Project"
  },
  {
    "startMs": 96120,
    "text": "search is, um, basically projects in Legora is, like, the, the unit of work that you have. So if, um, let's say you are SpaceX and you want to acquire Cursor, then that would be one project with your law firm. Uh, and so they would go into Legora, and they would upload all these documents, and the law firm that helped you would go through all of the employment agreements, all of the contracts with suppliers. Um, I know Cursor is using, uh, TurboPuffer, so maybe there's a contract there they'd look at. Um, but"
  },
  {
    "startMs": 125940,
    "text": "basically you do the search confined to a project, and projects can be tens of documents to millions of documents. The other use case is legal research, and legal research is sort of a deep research-style workload where we'll search across tons of laws, previous cases, regulations, et cetera, et cetera, and people use this to answer questions such as, like, \"How do I handle this specific thing?\" And they'll also use it for litigation. Maybe they want to sue someone or maybe they are getting sued, and they'll use legal research to support and help their case."
  },
  {
    "startMs": 156720,
    "text": "So if we start at number one, project search. We've been through a little bit of a ride here on how we do search, um, starting at, you know, hundreds of thousands of documents all the way into two billions of documents, and we've tried a lot of different things. So first we start, started with a very, very simple one, which is just a single Elasticsearch cluster for all of our search workloads. Um, that worked relatively well. It was sort of a simple setup. All of the tenants, you know, our clients, our users,"
  },
  {
    "startMs": 187040,
    "text": "would be on one big blob storage, where we store the raw documents, and on one big Elasticsearch, where we would do all of their searching, the indexing and the searching. Super simple. Worked relatively well initially. Then we wanted to enter the land of the free, and, uh, we got some new requirements. They... You know, Americans only want processing to happen within the US, and Europeans only want it to happen within the EU, and Australians only want it to happen within Australia. And so we had to s- basically, um... Well, here's a scaling"
  },
  {
    "startMs": 216980,
    "text": "graph. We had to move to multiple Elasticsearches, and what we actually did was we took the entire setup, and we just basically iterated over the set that is EU, US, and Asia Pacific, and so we just had this, like, multiplied by three or four. Kind of annoying, lots of overhead, but it, uh, got us to where we needed to be."
  },
  {
    "startMs": 237860,
    "text": "Then the next iteration of the story is enterprise. So really big banks, the biggest law firms in the world, they have really annoying requirements. And number one they have is they'll ask for full physical isolation of all of their data. There's probably a little bit of a war on what physical  you know, isolation actually means, but essentially it means they want their own database. They also want customer-managed encryption keys, and what that means is they basically have a key vault thing where they have an encryption key, and they give us"
  },
  {
    "startMs": 267900,
    "text": "access to read the key, and we then use that key to, uh, encrypt and decrypt all of their data at rest. And what that gives them is they can just revoke our access to their key, and then we can't decrypt their data anymore, and so it's safe. And so in a way, that gives enterprises a lot of control over all of their data 'cause they control, you know, the key to, to reading it."
  },
  {
    "startMs": 288220,
    "text": "So we moved from Elasticsearch to Postgres, and I imagine a bunch of you guys are like, \"Why would you ever put your vectors into Postgres?\"  Um, it actually works surprisingly well, and the reason that we did this was we were already using Pro- Postgres for OLTP workloads. And, uh, so we, we sort of already had to do this split of multiple Postgreses and multiple blobs. And so it was really easy for us to try to shift all of our search into Postgres as well 'cause then we only have one system. So the setup here was pgvector, uh,"
  },
  {
    "startMs": 318160,
    "text": "specifically diskANN, tsvector for the search, so not BM25, which was, you know, we lost a little bit of, uh, retrieval performance there. And what we'd do is we would partition the table where we would store all the document chunks. We'd partition it super aggressively, like 4,000 partitions, and then each project, we'd basically hash the project key, and we'd bin pack them into the partitions. That actually worked relatively well, but it was expensive, and search performance wasn't super, super good. And what happened was when we scaled"
  },
  {
    "startMs": 348100,
    "text": "a lot, uh, everything just broke and exploded. And so what happened was the... Basically, you can imagine, like, you have a bunch of projects, and some of them you spin up a project, you work on it, and then you close it, and you basically never go back to it again. And we have a bunch of those where, like, they never get queried, and we have a bunch that get queried all the time 'cause they're super active projects. And when we pack them into partitions, the cold ones and the hot ones would land on the same ones, and the partitions would get really, really big. And so when we queried them, they would... Postgres would pull the partition, put it into memory, we'd do the"
  },
  {
    "startMs": 377912,
    "text": "stuff, and then we'd query another partition, and another partition, and it would essentially thrash the cache all the time. And what that meant was our latencies would spike. So we went from, like, search and ingestion P99 of a hundred milliseconds into twenty seconds, which you can imagine is a really bad user experience."
  },
  {
    "startMs": 397972,
    "text": "So then we went to Turbopuffer in about, you know, when we were about, I think, four hundred million documents, something like that."
  },
  {
    "startMs": 405852,
    "text": "And what we did with Turbopuffer was we did one namespace per project. And the advantages of moving to Turbopuffer is we got BM25, real BM25, much better, uh, relevancy, much better latencies, and it was extremely simple to operate 'cause we could just have a single Turbopuffer cluster. You know, we didn't have to have a bunch of different ones like with Postgres, and it would, since it's blob-based, it could just query the blobs that we had anyway. And so much lower cost, and it was extremely simple to operate,"
  },
  {
    "startMs": 436312,
    "text": "and we didn't have this problem with the partitions 'cause if a project's not used, it's just in blob. And so it's really easy. And Simon can talk a bit more about why that works so well."
  },
  {
    "startMs": 448652,
    "text": "Yeah. So Legora has some of... And, and legal in general has... By the way, if, if, uh, Jake and I s- have similar accents and maybe even look a bit similar, it's because we're both Danish. Um, Turbopuffer has a very, has a particular architecture that supports these kinds of very regulated environments really, really well. But in order to understand that, we have to understand what kind of search engine is Turbopuffer. Why is it different than the ones that they used in the past? Since the very beginning of Turbopuffer, um, the design has more or less been the same. Uh,"
  },
  {
    "startMs": 478572,
    "text": "there may be changes in the future, but the design has stood the test of time. When you do a write to Turbopuffer, we write directly to object storage. There is no, like, disk replication, there's no Paxos, there's, there's none of that. Direct to S3. That's the fundamental trade-off in Turbopuffer, right? Hundreds of milliseconds. If you're like Shopify and doing inventory, uh, reservations for a Kylie Jenner flash sale, not gonna work. Very, very good for search because generally when you're doing search, doing a slow"
  },
  {
    "startMs": 508192,
    "text": "write is fine, um, as long as the read performance is, is adaptable and good. So that's what happens on write. It just goes in the write-ahead log. You can imagine you write one.json, two.json, three.json. Obviously, it's a da- it's, it's a database, so it's not JSON, but for illustrative purposes, that's what happens. And in the background, we build the vector indexes, the text indexes, the columnar indexes, and so on to satisfy the queries that, that Jake and other customers have. Um, so then at query time, we can go in and then, um, the query"
  },
  {
    "startMs": 538352,
    "text": "reaches some namespace, and namespace is kind of our concept of a, of a table. Um, you can think of it as a directory on S3 that's isolated from everything else. We go to the node that is most likely to have it. It could go to any node, right? It could go to every single node, and they're all read replicas, but it would... we go with some affinity to the node that has the highest probability of having it in cache. We check the memory cache for any objects, NVMe SSD cache, and then finally to object storage. Everything in Turbopuffer is optimized around doing as much work in as few round trips as possible, right? S3 has a P99,"
  },
  {
    "startMs": 568732,
    "text": "um, on a, like, one megabyte blob size of around, uh, two hundred milliseconds, so you wanna do as few round trips as possible, right? Ideally, you do around three. And everything in Turbopuffer, the database, is... Oh, I'm gonna need your fingerprint."
  },
  {
    "startMs": 581952,
    "text": "You got it."
  },
  {
    "startMs": 582512,
    "text": "Um, everything in, in Turbopuffer is designed around minimizing the number of round trips. This is also amazing for modern dri- disks. If you do a lot of concurrency in few round trips, you utilize them optimally, and everything in Turbopuffer is designed around this. So why is this so good for a company like Legora? Well, object storage native, if you design it around the atomic unit of separation being the namespace or the table, every single table could be encrypted with a different key. Every single namespace could be in a different bucket. We have"
  },
  {
    "startMs": 612212,
    "text": "customers that have, um, thousands of buckets that they have namespaces in so that their customers get the warm IT fuzzies of having the bucket in their own cloud account. They can also be encrypted with their own keys. You can share buckets. You can, you can do whatever configuration that you need at the namespace level. You can re-encrypt with different keys, you can move them around, um, and you can re-encrypt with other keys. Um, for Legora in particular, this was really important for this full physical separation, right? An encryption"
  },
  {
    "startMs": 641892,
    "text": "separation. All of the namespaces needed to be physically at rest with different keys and as separate as possible. S3, GCS, Azure Blob Storage, they pass that. Um, and the other parts of the hierarchy also. Except the NVMe SSD cache, because in the SSD cache, we consider that to be volatile like memory, um, but your customers did not.  So in the, uh... So what we did was that we thought we were gonna implement encryption into the disk cache, but instead we just disabled the disk cache"
  },
  {
    "startMs": 671932,
    "text": "and saw how it fared. And the performance of Turbopuffer, even without the disk cache with just the memory cache, was so good that we just kept it that way for some of the Legora workloads where we couldn't have the disk cache for multi-tenancy. Turbopuffer will support that in the future. But it just goes to show the, um, natural point where Turbopuffer allows these encryption and storage and separation to become fully multi-tenancy, uh, native. I'll hand it back to you on what happened then."
  },
  {
    "startMs": 699232,
    "text": "And then, drum roll please, latencies looked like this. Um, is my mic working?"
  },
  {
    "startMs": 707072,
    "text": "No."
  },
  {
    "startMs": 708272,
    "text": "No? Could I... Or I'll start screaming really loudly. Um, it speaks for itself if you can't hear me. Okay. Um, latencies improved in order of magnitude basically, and these are median latencies, so P99 were even better. Um, so obviously this is a huge thing when you're doing-- I mean, one thing is if you're doing a single sort of RAG-style thing, but if you have an agent that does twenty queries, a hundred queries, these really, really add up. So that was on the project side, and then a more recent thing is legal research."
  },
  {
    "startMs": 738212,
    "text": "So legal research, um, is a kind of a difficult problem, and the reason it's difficult is that, um, the corpus is extremely big, so we're racing towards ten billion vectors, and we're growing extremely fast. We also have quite high read, um, so QPS can spike a lot 'cause we do a lot of fan-out. Like, if you do a, a, a sort of legal research query, we will fan it out into a bunch of different queries, and we'll keep going. And the reason we do that is we need this, um, heavy filtering 'cause essentially"
  },
  {
    "startMs": 768192,
    "text": "it's a-- kinda like a graph for a few different reasons. Firstly, it's, um, hierarchical. You know, you have cities, and you have counties, and you have states, and you have federal law, and it's the same all around the world. Um, and so you need to respect that authoritative sort of hierarchy. There's also some temporal validity, so one judge might overrule a decision that's been made somewhere else, and you need to also respect that and figure that out. And then sometimes there's even, like, a new regulation that has exemptions or special cases of an old regulation, and"
  },
  {
    "startMs": 798132,
    "text": "so if you're finding this one, you need to find all the other ones as well. So you can imagine that it, uh, sort of explodes the search."
  },
  {
    "startMs": 806532,
    "text": "And so we started on Elasticsearch for this, but also moving to TurboPuffer. Um, Elasticsearch just got extremely expensive 'cause we have to have everything there. But, um, with TurboPuffer, we can basically, uh, take different jurisdictions, and we can make them namespaces in TurboPuffer, and that means some of them... Here's an example where, like, you have the EU. That gets queried all the time. That's super hot. And some of them, let's say Danish law 'cause we're Danish, no one cares really. It's such a small country, so, like, it doesn't really get queried, and so that can just stay on Blob, and that's fine."
  },
  {
    "startMs": 836852,
    "text": "Um, and because it's sort of a deep research-style workload, if there's five hundred milliseconds latency to fetch that cold Blob, that's okay. That's fine. It's not really a big problem. So the way that TurboPuffer is, is designed lends itself super well to this super long scale of, like, cold, weird namespaces and a few that are really, really hot. Um, yeah, and Simon wants to talk more about that."
  },
  {
    "startMs": 860192,
    "text": "Yeah. So, um, um, I was, I was talking about why the company is called TurboPuffer at another, uh, talk here earlier today. But, uh, one of the other explanations of the name of TurboPuffer is that it's about puffing into the different memory hierarchies and really mastering when data should be in particular memory hierarchies. So you can think about it here, right, of something like the EU law might be more or less part of almost every one of the legal research, uh, queries, right? So that probably sits closer"
  },
  {
    "startMs": 890232,
    "text": "to NVMe SSDs than memory, right? The economics kinda change as you move up and down this hierarchy. Um, in, in memory, you want things that are queried a lot, right? Then the economics of memory are great. NVMe SSDs can-- you can do a lot of things directly on them, but the economics change as you move up and down this boundary. The latency changes, and the way that the database is architected to take advantage of it in terms of round trips versus random versus sequential all changes as you navigate this hierarchy. TurboPuffer is a database that is really designed around the memory"
  },
  {
    "startMs": 920012,
    "text": "hierarchy, and all of the smarts in TurboPuffer is that all of these namespaces are puffed in and out, um, of the cache. You can think of this as we wanna spend as much time, have as much data pushed as far down in this hierarchy as possible to get the, the best, um, performance cost ratios. So how does that apply to search? Well, for something like vector search, for example, there's two fundamental ways to do vector search. One is to navigate it, basically design a graph. The problem with a graph on something like object storage or disk, again, we wanna"
  },
  {
    "startMs": 949992,
    "text": "have things as far down that memory hierarchy as possible. The problem with a graph, this is not a graph, this is a tree, um, but in a graph, you have to navigate from the center of the graph. So-- And then every time you navigate through these nodes, you're doing two hundred millisecond P99 to S3, right? And so you're trying to shrink the diameter of the graph. You're trying to do all these tricks to make the graph. But fundamentally, you're at odds with the fact that a graph is about a random sequential trade-off that you have in memory and in registers but not further down the memory hierarchy."
  },
  {
    "startMs": 980592,
    "text": "The way TurboPuffer does it is organize it into clusters, right? Vectors you can think of in two dimensions just as point is a massive coordinate system, and we can organize them into clusters. TurboPuffer then creates clusters of clusters and clusters of clusters of clusters to essentially organize all of the vector data in a tree. You can basically think of TurboPuffer as a very, very complicated B-tree,  right? Because it's a tree on this geometry of this entire space and the clustering of it in an approximate way. Now, the root centroids further up"
  },
  {
    "startMs": 1010552,
    "text": "the tree, you can imagine, are part of every single time you search, right? They're-- We're always trying to figure out which clusters that we're in, and we're always looking at the upper levels of the tree. So they're gonna be further up the memory hierarchy, right? Closer to the registers, m-almost all in DRAM. Now, the leaves that have all of the actual legal cases of whatever long document it could be, it could be images, all of that is probably gonna be on SSDs with that single one-millisecond round trip at the end. It doesn't make sense to have all that puffed into DRAM. This is fundamentally the cheapest"
  },
  {
    "startMs": 1039992,
    "text": "way that you can run a database, period. So for something like Legora or even web search, which is in the hundreds of billion or tens of billions, depending on how much of the web you've scraped, this is fundamentally the cheapest way to do it. And we have customers that are indexing massive parts of the entire web into TurboPuffer, which is really also a part of what legal research is. Full text is also, also really respectful of the memory hierarchies. The way that text search works is essentially you can think of it as a HashMap. You have a"
  },
  {
    "startMs": 1070032,
    "text": "big document, and then you take every single one of the tokens, and you put them into the key in the HashMap. The value in the HashMap is some set with all of the document IDs that has that term. So then if you search for New York population, you're finding those three places in the HashMap, and then you're taking the three sets and doing an intersects on the- intersect on the sets. While you're intersecting, you're also trying to do some kind of scoring, right? A document that has York in it is probably more valuable than a document that has New in it because York is a more"
  },
  {
    "startMs": 1099872,
    "text": "rare word. When people say BM25, this is the scoring that they're referring to. The art of full-text search is, one, we wanna minimize the number of round trips. So first you download the parts of the dictionary that are relevant, round trip one, maybe a round trip one before that to index into where the parts of the term- terms are. And then the second round trip is to get these massive lists. Try to make the lists as small as possible by compressing them. But also while you're doing the text search, you're trying to minimize the amount, again, of memory bandwidth that you"
  },
  {
    "startMs": 1129792,
    "text": "want to intersect these lists. You can probably imagine that at some point, there's a point where you've seen so many documents with population and York that have much higher scores that documents that just have New in it are irrelevant anymore. This is like a mega crash course in how text search works. And counterintuitively to most people, text search at web scale is more difficult and more computationally expensive than doing vector search. I'll hand it over to you."
  },
  {
    "startMs": 1156852,
    "text": "Cool. So key learnings from, um, what you heard today. Retrieval is extremely important to Legora. Um, it's key to legal reasoning. TurboPuffer really excels when, uh... for us 'cause, uh, it makes it extremely easy to operate. We have seventy-plus tenants. We have a hundred. We have two hundred tenants. You know, if we had to have separate Elasticsearch databases for each of these, it would be just hell. Um, but we can do this natively with TurboPuffer with, uh, data residency and CMAC, et cetera, et cetera."
  },
  {
    "startMs": 1187392,
    "text": "And then it's extremely cost efficient generally when you have these types of workflows, um, or workloads that we do where there's a long tail of cold indices basically that you don't need to query so much, and you don't-- you, you're sort of... You're okay paying the small, uh, latency cost for it. So now with, um, TurboPuffer and four seconds to go, now we can focus on making Legora. We can focus on the product, making it really, really great, not on scalability and infra. And also David, our CFO, is really happy about the cost, so it's great. Thanks, everyone."
  }
]
