Research engineer @scale_AI

San Francisco, CA
Joined September 2015
Replying to @PennJenks
Getting the GPT2 124M to 2.906 val loss in about 20 hours with a TPU v3-32 pod slice.
15
2,197
Just as graphics rendering enabled neural nets and AI through the development of GPUs, AI will enable scientific discovery through massive simulations utilizing data centers originally built for training and inference.
1
464
Not saying Meta should be using TPUs, but you probably should be.
1
6
635
doesn't seem trustworthy. time will tell.
1
320
was just looking at cursor ais blog post "instant apply" (now removed). I'm wondering - did they waste a bunch of time - and should have just used vllm ngram decoding?
Up to 3x faster LLM generation with no extra resources/requirements - ngram speculation has landed in 🤗 transformers! 🏎️💨 All you need to do is to add `prompt_lookup_num_tokens=10` to your `generate` call and you'll get faster LLMs🔥 🧵How does it work?
284
love this energy
Replying to @_marleyx
We decided to measure it ourselves. Turns out you can buy a human skull online from skullsunlimited.com...
1
175
alcohol is just high temperature sampling
5
152
I can't believe google released mansplaining as a product
2
164
Enjoy attention? Want to make it ~18% faster? Try out Sigmoid Attention. We replace the traditional softmax in attention with a sigmoid and a constant (not learned) scalar bias based on the sequence length. Paper: arxiv.org/abs/2409.04431 Code: github.com/apple/ml-sigmoid-… This was an amazing collaboration with some great researchers at Apple: Federico Danieli, @EeshanDhekane, @FlorisWeers, @danbusbridge, @PierreAblin, Tatiana Likhomanenko, Jagrit Digani, Zijin Gu, @AmitisShidani1 and Russ Webb. More details in 🧵below.
15
163
16
826
128,511
dear llm providers, please allow tools to return images. thank you. @OpenAI @AnthropicAI @GoogleDeepMind
1
3
189
Penn Jenks retweeted
If you know Torch, I think you can code for GPU now with OpenAI's Triton language. We made some puzzles to help you rewire your brain. Starts easy, but gets quickly to fun modern models like FlashAttention and GPT-Q. Good luck! github.com/srush/Triton-Puzz…
14
193
9
1,352
159,740
Fine-tuning the LLaMA-2-Chat model may degrade its original capabilities (arxiv.org/abs/2401.03129). But here's a lifeline: Chat Vector (arxiv.org/abs/2310.04799) keeps a chat model's original capability (it also works on Mistral). Recommend to everyone fine-tuning their LLMs.
2
79
5
357
38,958
If you're interested in evaluating LLMs in Korean but are put off by the practice of translating evaluation data into other languages, try these benchmarks! HAE-RAE focuses on Korean-specific knowledge and culture. K-MMLU is a replication of MMLU using Korean national exams.
🔥 Excited to share: 1️⃣ HAE-RAE Bench: Evaluation of Korean Knowledge in Language Models (LREC-COLING 2024) 2️⃣ KMMLU: Measuring Massive Multitask Language Understanding in Korean (Under Review) 📢 TL;DR -> Together with @AiEleuther we improve evaluation practices for Korean language by releasing two datasets HAE-RAE Bench and KMMLU. [1/4]
6
18
4,908
Penn Jenks retweeted
Happy to see that our work on diffusion model objectives (arxiv.org/abs/2303.00848) is starting to get noticed, with e.g. Stable Diffusion 3 (arxiv.org/abs/2403.03206) building on our result that the flow matching objective can be simply understood as a special case of diffusion.
5
41
2
366
47,150
Do you want to train massive deep learning models with ease? Our 10 new tutorial notebooks of our popular UvA DL course show you how, implementing data, pipeline and tensor parallelism (and more) from scratch in JAX+Flax! 🚀🚀 Check them out here: uvadlc-notebooks.readthedocs… 🧵 1/11
10
116
9
611
71,987
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,193
Penn Jenks retweeted
New Griffin paper is really interesting and contains a lot of implementation details arxiv.org/abs/2402.19427 . Implementation is in Pallas which is a Jax like frontend to Triton/TPU lowering. They show that Associative Scan is inherently worse than Linear Scan in this context. (not sure if this is TPU specific.)
4
39
1
280
35,047
Penn Jenks retweeted
LLM planning methods, such as tree search, are critical for complex problem solving, but their practical utility can depend on the discriminator used with them. Check out our new findings: arxiv.org/abs/2402.10890 (1/6)
6
44
9
184
84,931
Penn Jenks retweeted
Let's implement Mamba in Triton. (srush.github.io/annotated-ma…) A gentle, (but mildly obsessive) tutorial notebook about GPU programming in Triton. We're getting close to mere mortals being able to do this 😂
10
155
3
1,039
102,967
I find it deeply confusing that nobody has gotten around to doing this yet. Super low hanging fruit and requires ~ zero compute too! Mostly it requires a high-quality human study. #stealmyidea
Replying to @BlancheMinerva
FID is largely meaningless for T2I models because it completely ignores the pairing of the prompt and the output. Develop a "multimodal FID" that is aware of both text and image embeddings based on this paper arxiv.org/abs/2103.11521
1
2
31
5,626
Blog post about how to scale training runs to highly distributed settings (i.e., large batch sizes)! Empirical insights from my long-ago work on stochastic differential equations (SDEs). Written to be accessible - give it a shot! cs.princeton.edu/~smalladi/b…
7
72
4
379
81,620