Aaryan Singhal's starred repositories
schedule_free
Schedule-Free Optimization in PyTorch
ThunderKittens
Tile primitives for speedy kernels
consistency-policy
[RSS 2024] Consistency Policy: Accelerated Visuomotor Policies via Consistency Distillation
nanoGPT-TK
The simplest, fastest repository for training/finetuning medium-sized GPTs. Now, with kittens!
madrona_escape_room_pixact
co-optimizing the design and execution of ml architectures that enables end-to-end (raw pixel inputs, agent action outputs) rl algorithms to quickly learn effective game-playing policies on the madrona game engine