@ContextualAI MTS | Quantum Information PhD

Bay Area
Joined June 2011
Hard to believe it’s barely been a year since @douwekiela called to order our very first all hands, then @apsdehal and I spent hours in a tiny room with a whiteboard laying out the technical vision for what we were about to do. So proud of the team we’ve built and what’s to come!
We’re excited to share today that we’ve raised $80M in Series A funding to accelerate our mission to change the way the world works through AI. Read more at our blogpost: contextual.ai/news/announcin…
1
1
12
1,582
Another great example of how we frame the problems of enterprise ai from a systems perspective... and this component is a key player 😎 (which is also SOTA by itself 🙌
AI struggles with messy, conflicting, ever-changing data. Today's AI ranking methods can't prioritize clearly, because they lack human guidance. Introducing the world's first instruction-following, SOTA reranker! Give our reranker instructions to control exactly how it ranks: • “Prioritize recent documents” • “Prefer PDFs over other sources” • “The boss is always right” Can’t wait to see what people build with it!
5
112
Here, have a truly open MoE 🤲. More amazing work by @Muennighoff! Tons to dive into here.
Releasing OLMoE - the first good Mixture-of-Experts LLM that's 100% open-source - 1B active, 7B total params for 5T tokens - Best small LLM & matches more costly ones like Gemma, Llama - Open Model/Data/Code/Logs + lots of analysis & experiments 📜arxiv.org/abs/2409.02060 🧵1/9
6
551
Casey A. Fitzpatrick retweeted
Here is a new Machine Learning Engineering chapter: Network debug github.com/stas00/ml-enginee… The intention is to help non-network engineers to figure out how to resolve common problems around multi-gpu and multi-node collectives networking - it's heavily NCCL-biased at the moment. Will extend with RCCL and others when I get access to those. Your feedback and corrections are always welcome.
4
40
1
171
9,434
ty @charles_irl and gang
The kind folks from @modal_labs have just shared with me this 10-100x faster drop in replacement for pip github.com/astral-sh/uv If you want a much faster CI startup switch to uv now! To use you just add `uv` before `pip` and everything else is the same, so: pip install uv uv pip install -e . uv pip install torch uv pip compile ... etc.
3
144
Casey A. Fitzpatrick retweeted
Maximizing expected human utility (i.e., KTO) is the natural way to do alignment. Cool to see how well this works even in diffusion models.
Aligning Diffusion Models by Optimizing Human Utility We present Diffusion-KTO, a novel approach for aligning text-to-image diffusion models by formulating the alignment objective as the maximization of expected human utility. Since this objective applies to each
5
22
5,936
Casey A. Fitzpatrick retweeted
from @douwekiela (a pioneer of Retrieval-Augmented Generation) on the value of end-to-end co-optimization of systems (language models + retrievers):
Today, we’re excited to announce RAG 2.0, our end-to-end system for developing production-grade AI. Using RAG 2.0, we’ve created Contextual Language Models (CLMs), which achieve state-of-the-art performance on a variety of industry benchmarks. CLMs outperform strong RAG baselines built using GPT-4 and top open-source models like Mixtral, according to our research and customers. Read more in our blog post: rag2.ai
5
28
11,066
🤩
Not everyone gets to call their new system RAG 2.0 — exciting announcement by the @ContextualAI team, who have been thinking about these problems for many years.
120
rip FRAG 🪦 we'll never forget the many meetup demos
2023: RAG vs no RAG 2024: RAG 2.0 vs frozen RAG As AI moves into production, simple RAG systems are not enough. @douwekiela & @ContextualAI show that with data.
5
91
Last year I joined Contextual AI to focus on designing and building production-grade AI systems, from first principles, focused on real world workflows not demos. Today I'm excited to share some of what we've done so far!
Today, we’re excited to announce RAG 2.0, our end-to-end system for developing production-grade AI. Using RAG 2.0, we’ve created Contextual Language Models (CLMs), which achieve state-of-the-art performance on a variety of industry benchmarks. CLMs outperform strong RAG baselines built using GPT-4 and top open-source models like Mixtral, according to our research and customers. Read more in our blog post: rag2.ai
1
10
280
👋hiiii we’re here to play!
Excited to share a new model with @ContextualAI that tops the AlpacaEval 2.0 leaderboard! How did we manage to rank higher than models like GPT4, Claude 3 and Mistral Medium? Enter iterative alignment… 🧵
3
127
Casey A. Fitzpatrick retweeted
The Orca-Math paper does a comparison of DPO and KTO for mathematical reasoning, finding that KTO is slightly better when all data is used and 25+ pts better when you have fewer positive examples than negative examples.
Microsoft presents Orca-Math Unlocking the potential of SLMs in Grade School Math Mathematical word problem-solving has long been recognized as a complex task for small language models (SLMs). A recent study hypothesized that the smallest model size, needed to achieve over 80% accuracy on the GSM8K benchmark, is 34 billion parameters. To reach this level of performance with smaller models, researcher often train SLMs to generate Python code or use tools to help avoid calculation errors. Additionally, they employ ensembling, where outputs of up to 100 model runs are combined to arrive at a more accurate result. Result selection is done using consensus, majority vote or a separate a verifier model used in conjunction with the SLM. Ensembling provides a substantial boost in accuracy but at a significant cost increase with multiple calls to the model (e.g., Phi-GSM uses top-48 to boost the performance from 68.2 to 81.5). In this work, we present Orca-Math, a 7-billion-parameter SLM based on the Mistral-7B, which achieves 86.81% on GSM8k without the need for multiple model calls or the use of verifiers, code execution or any other external tools. Our approach has the following key elements: (1) A high quality synthetic dataset of 200K math problems created using a multi-agent setup where agents collaborate to create the data, (2) An iterative learning techniques that enables the SLM to practice solving problems, receive feedback on its solutions and learn from preference pairs incorporating the SLM solutions and the feedback. When trained with Supervised Fine-Tuning alone, Orca-Math achieves 81.50% on GSM8k pass@1 metric. With iterative preference learning, Orca-Math achieves 86.81% pass@1. Orca-Math surpasses the performance of significantly larger models such as LLAMA-2-70B, WizardMath-70B, Gemini-Pro, ChatGPT-3.5. It also significantly outperforms other smaller models while using much smaller data (hundreds of thousands vs. millions of problems).
2
23
1
120
33,192
Casey A. Fitzpatrick retweeted
I'm glad that a lot more people understand the key ideas behind ColBERT and DSPy now. My only remaining goal is to make sure people can also say them correctly; both are quite tricky😆 * Col-BAIR (it's "the late" interaction retriever, get it?) * Dee-Ess-Pie (like num-pie)
18
11
1
183
33,796
😊 🙏
I know everyone is excited about Mixtral and the new Hyena models - but @ContextualAI just dropped a pile of cool new models and a new alignment framework contextual.ai/better-cheaper…
3
164
Casey A. Fitzpatrick retweeted
I believe strongly that: 1) The best products that will emerge from this moment are “full stack”, with teams training their own models, and the models & UI informing one another. 2) This requires researchers who care deeply about what’s best for the product, including data
19
24
6
261
85,320
🌠🌠🌠
We're proud to be a 2023 #IA40 Intelligent Applications Rising Star winner!
80
Stas has done some incredible work. We’re thrilled to have him on the team to help us take retrieval systems to the next level!
I'm super excited to start working at @contextualai where I will be training LLMs w/ Retrieval to help businesses deploy AI that overcomes hallucination, keeps data up-to-date and runs much faster inference. If you're new to Contextual.AI, see: contextual.ai/announcing-nex… Applied ML here I come!
1
1
5
807
Casey A. Fitzpatrick retweeted
It was great to see Contextual AI in @ThomasOrTK's keynote at #GoogleCloudNext. We’re excited to partner and build the next generation of language models. cloud.google.com/blog/topics…
1
7
20
3,147
Training ml models is good for ml models 🤔Probably some cool actual-cognitive-science lurking here behind the idea…?
Garbage in and Garbage out analogy here w.r.t. Retrieval strongly resonates with me. Generator as of now doesn't know what retriever is going to send and has no way of giving any feedback on what was useful. ➰ The only feedback loop right now is through humans tuning/trying out various retrieval hacks based on their use case. @ContextualAI we are solving these issues by thinking through first principles for enterprises specifically.
1
1
205