New HYBRID AI Model Just SHOCKED The Open-Source World - JAMBA 1.5

AI21 Labs released Jamba 1.5, a hybrid model combining transformers with state space models to handle longer context faster and cheaper. The Mini version runs 140k tokens on a single GPU, the Large handles 256k tokens on eight GPUs, and both beat other models on standard benchmarks.
Chapters
So AI21 Labs, the brains behind the Jurassic language models, has just dropped two brand new open source LLMs called Jambo 1.5 Mini and Jamba 1.5 Large. And these models are designed with a unique hybrid architecture that incorporates cutting edge techniques to enhance AI performance. And since they're open source, you can try them out yourself on platforms like Hugging Face or run them on cloud services like Google Cloud, Vertex and AI, Microsoft Azure and Nvidia Nim. Definitely worth checking out. Alright, so what's this hybrid architecture all about? Okay, let's break it down in simple terms. Most of the language models, you know, like the ones used in ChatGPT, are based on the transformer architecture. These models are awesome for a lot of tasks, but they've got this one big limitation. They struggle when it comes to handling really large context windows. Think about when you're trying to process a super long document or a full transcript from a long meeting. Regular transformers get kind of bogged down because they have to deal with all that data at once. And that's where these new Jamba models from AI21 Labs come into play with a totally new game changing approach. So AI21 has cooked up this new hybrid architecture they're calling the SSM Transformer. Now what's cool about this is it combines the classic transformer model with something called a structured state space model or ssm. The SSM is built on some older, more efficient techniques like neural networks and convolutional neural networks. Basically, these are better at handling computations efficiently. So by using this mix, the Jamba models can handle much longer sequences of data without slowing down. That's a massive win for tasks that need a lot of context, like if you're doing some complex generative AI reasoning or trying to summarize a super long document. Now why is handling a long context window such a big deal? Well, think about it. When you're using AI for real world applications, especially in businesses, you're often dealing with complex tasks. Maybe you're analyzing long meeting transcripts, or summarizing a giant policy document, or even running a chatbot that needs to remember a lot of past conversations. The ability to process large amounts of context efficiently means these models can give you more accurate and meaningful responses. Or Dagan, the VP of Product at AI21 Labs, actually nailed it when he said an AI model that can effectively handle long context is crucial for many enterprise generative AI applications. And he's right. Without this ability, AI models often tend to hallucinate or just make stuff up because they're missing out on important information. But with the Jamba models and their unique architecture, they can keep more relevant info and memory, leading to way better outputs and less need for repetitive data processing. And, and you know what that means? Better quality and lower cost. Alright, let's get into the nuts and bolts of what makes this hybrid architecture so efficient. So there's one part of the model called Mamba which is actually very important. It's developed with insights from researchers at Carnegie Mellon and Princeton. And it has a much lower memory footprint and a more efficient attention mechanism than your typical transformer. This means it can handle longer context windows with ease, unlike transformers, which have to look at the entire context every single time. So slowing things down, Mamba keeps a smaller state that gets updated as it processes the data. This makes it way faster and less resource intensive. Now, you might be wondering, how do these models actually perform? Well, AI21 Labs didn't just hype them up, they put them to the test. They created a new benchmark called Ruler to evaluate the models on tasks like multi hop, tracing, retrieval, aggregation and question answering. And guess what? The Jamba models came out on top consistent, consistently outperforming other models like Llama 3.170 B, Llama 3.1, 405B and Mistral Large 2. On the arena Hard benchmark, which is all about testing models on really tough tasks, Jamba 1.5 Mini and Large outperformed some of the biggest names in AI. Jamba 1.5 Mini scored an impressive 46.1, beating models like Mixtral 8x22B and Command R, while Jambo 1.5 Large scored a whopping 65.4, outshining even the big guns like Llama 3.170B and 405B. One of the standout features of these models is their speed. In enterprise applications, speed is everything. Whether you're running a customer support chatbot or an AI powered virtual assistant, the model needs to respond quickly and efficiently. The Jamba 1.5 models are reportedly up to 2.5 times faster on long contexts than their comparison competitors. So not only are they powerful, but they're also super practical for high scale operations. And it's not just about speed. The Mamba component in these models allows them to operate with a lower memory footprint, meaning they're not as demanding on hardware. For example, Jamba 1.5 mini can handle context lengths up to 140,000 tokens on a single GPU. That's huge for developers looking to deploy these models without needing a massive infrastructure. Alright, here's where it gets even cooler. To make these massive models more efficient, AI21 Labs developed a new quantization technique called Experts Int8. Now I know that might sound a bit technical, but here's the gist of it. Quantization is basically a way to reduce the precision of the numbers used in the model's computations. This can save on memory and computational costs without really sacrificing quality. Experts Int8 is special because it specifically targets the weights in the mixture of Experts or MOE layers of the model. These layers account for about 85% of the model's weights in many cases. By quantizing these weights to an eight bit precision format and then de quantizing them directly inside the GPU during runtime, AI21 Labs managed to cut down the model size and speed up its processing. The result? Jamba 1.5 large can fit on a single 8 GPU node while still using its full context length of 256k. This makes Jamba one of the most resource efficient models out there, especially if you're working with limited hardware now. Besides English, these models also support multiple languages including Spanish, French, Portuguese, Italian, Dutch, German, Arabic and Hebrew, which makes them super versatile for global applications. And here's a cherry on top, AI21 labs made these models developer friendly. Both Jamba 1.5 mini and large come with built in support for structured JSON, output, function calling and even citation generation. This means you can use them to create more sophisticated AI applications that can perform tasks like calling external tools, digesting structured documents and providing reliable references, all of which are super useful in enterprise settings. One of the coolest things about Jamba 1.5 is AI21 Labs commitment to keeping these models open. They're released under the JAMBA Open Model License, which means developers, researchers and businesses can experiment with them freely and with availability on multiple platforms and cloud partners like AI21 Studio, Google Cloud, Microsoft Azure, Nvidia NIM, and soon on Amazon, Bedrock, Databricks Marketplace and and more. You've got tons of options for how you want to deploy and experiment with these models. Looking ahead, it's pretty clear that AI models that can handle extensive context windows are going to be a big deal in the future of AI. As Ordegin from AI21 Labs pointed out, these models are just better suited for complex, data heavy tasks that are becoming more common in enterprise settings. They're efficient, fast and versatile, making them a fantastic choice for developers and businesses looking to push the boundaries in AI. So if you haven't checked out Jamba 1.5 mini or large yet, now's the perfect time to dive in and see what these models can do for you. Alright, if you found this video helpful, smash that like button, hit subscribe and stay tuned for more updates on the latest in AI tech. Thanks for watching and I'll catch you in the next one.










