Sr. Deep Learning Engineer @NVIDIA Visual Gen AI | DL Algorithms
Prev. @latticeflowai Seervision @ETH @amzracing @motionaldrive @UniFreiburg @iiit_hyderabad VIT
Building generative models using libraries is great for getting started but the high-level APIs abstract away the details (usually where the devil is)
I implemented Torchsmith, an open-source package that implements generative models from scratch using basic PyTorch operations🧵
Scrolling the AI news timeline as a researcher feels like a teenager browsing Instagram:
"Everyone else has figured everything out!"
Reliable home robots imminent, 100× productivity AI agents, insane visual generation ...
Exciting, but anxiety-inducing. What am I doing? 😬
Applications change, but the principles are enduring. After a year's hard work led by @JCJesseLai, we are really excited to share this deep, systematic dive into the mathematical principles of diffusion models. This is a monograph we always wished we had.
New paper 📜: Tiny Recursion Model (TRM) is a recursive reasoning approach with a tiny 7M parameters neural network that obtains 45% on ARC-AGI-1 and 8% on ARC-AGI-2, beating most LLMs.
Blog: https://t.co/w5ZDsHDDPE
Code: https://t.co/7UgKuD9Yll
Paper: https://t.co/3m8ANhNMiw
(5/5) This thread is a concise version, check out the repo on GitHub for the full details
https://t.co/Yv5EwPpuw7
P.S.: Definitely recommend this way for anyone who wants to get a deeper understanding and see how to train and sample from such models
Building generative models using libraries is great for getting started but the high-level APIs abstract away the details (usually where the devil is)
I implemented Torchsmith, an open-source package that implements generative models from scratch using basic PyTorch operations🧵
it’s been an interesting ride watching the conventional nomenclature of machine learning gradually lose all meaning. there used to be TRAIN and TEST and everything was simple. now we train on the universe. and we test on the universe, too. are we gaming our benchmarks? are we extrapolating or interpolating? if a model is trained on the entire internet but generates a single novel sentence, is it just combining phrases from Reddit or writing something truly novel? feels like we lack the words to even describe the systems we’ve built.
We did a very careful study of 10 optimizers with no horse in the race. Despite all the excitement about Muon, Mars, Kron, Soap, etc., at the end of the day, if you tune the hyperparameters rigorously and scale up, the speedup over AdamW diminishes to only 10% :-( Experiments are made possible by Marin (https://t.co/UgEjGM0HPY); anyone developing new optimizers: please come try your method on this benchmark!
NVIDA chips are manufactured by TSMC, a Taiwanese company. They're created using EUV lithography machines manufactured by ASML, a Dutch company. These machines consist of >50% of German parts (by value), in particular ZEISS optics.
@giffmana@XiaohuaZhai@__kolesnikov__@_basilM Okay, so the values were obtained via hyper parameter search.
Not clear what you mean when referring to "prior".
> "This is because the bias term ensures that the training starts close to the prior"
@giffmana@XiaohuaZhai@__kolesnikov__@_basilM "We initialize t' and b to log 10 and −10 respectively. This makes sure the training starts roughly close to the prior ..."
Couldn't find more about how the two values were determined in the text.