
Research engineer @scale_AI
San Francisco, CA
Joined September 2015
- Tweets1.3K
- Following384
- Followers192
- Likes2.4K
Pinned Tweet
Replying to @PennJenks
Getting the GPT2 124M to 2.906 val loss in about 20 hours with a TPU v3-32 pod slice.
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.
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?
love this energy
Replying to @_marleyx
We decided to measure it ourselves.
Turns out you can buy a human skull online from skullsunlimited.com...
Penn Jenks retweeted
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.
dear llm providers, please allow tools to return images. thank you.
@OpenAI @AnthropicAI @GoogleDeepMind
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…
Penn Jenks retweeted
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.
Penn Jenks retweeted
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]
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.
Penn Jenks retweeted
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
Penn Jenks 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).
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.)
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)
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 😂
Penn Jenks retweeted
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
Penn Jenks retweeted
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…

































