Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

Scripts for fine-tuning Meta Llama with composable FSDP & PEFT methods to cover single/multi-node GPUs. Supports default & custom datasets for applications such as summarization and Q&A. Supporting…

Kolmogorov Arnold Networks

llama3 implementation one matrix multiplication at a time

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…

Neural Networks: Zero to Hero

Solve puzzles. Learn CUDA.

Official inference library for Mistral models

This repository contains demos I made with the Transformers library by HuggingFace.

Material for gpu-mode lectures

GaussianObject: High-Quality 3D Object Reconstruction from Four Views with Gaussian Splatting (SIGGRAPH Asia 2024, TOG)

All the projects related to Llama

small auto-grad engine inspired from Karpathy's micrograd and PyTorch

look how they massacred my boy

The code for "Diffusion Geometry" (2024).

Contains the code for the paper "MapInWild: A Remote Sensing Dataset to Answer the Question What Makes Nature Wild"