instructGOOSE. Implementation of Reinforcement Learning from Human Feedback (RLHF)
172toolformer. Implementation of Toolformer: Language Models Can Teach Themselves to Use Tools
146pipegoose. Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
87stable-diffusion-from-scratch. Implementation of Stable Diffusion from scratch [WORK IN PROGRESS]
22reinforcement-learning. Jupyter Notebook
10ai-notebooks. AI notebooks
5neuraltext. Decoding neural activity into text
5progen. Generating new proteins using language models
4flashattention. Implementation of FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness
4prodgpt. A production-ready training, evaluation and data pipeline
4muzero. Implement MuZero from scratch [WORK IN PROGRESS]
3fastgoose. A PyTorch implementation of Model Parallelism and ZeRO Optimizer
3nanoGPT-msamp. Integrate MS-AMP into nanoGPT (https://github.com/karpathy/nanoGPT)
3saycan. Implementation of SayCan: Grounding Language in Robotic Affordances
2homework. Notes
2snippets. snippets
2mousai. PyTorch Implementation of Moûsai: Text-to-Music Generation with Long-Context Latent Diffusion
2Megatron-LM. Ongoing research training transformer models at scale
1transformers-starcoder. Python
1hf-notebooks. Jupyter Notebook
1Jetfire-INT8Training. Jupyter Notebook
1megatron-tp. for debugging pipegoose
1fsdl-megatron. Code for FSDL Breaking down parallelism in Megatron-LM
1mq3-sensor. Python
1research-ideas. List of things that I'd love to work on
1pipegoose-training. Jupyter Notebook
1xrsrke.
1fsdl-website. Source for https://fullstackdeeplearning.com
1memit. Implementation of Mass Editing Memory in a Transformer
1elasticgoose. A fault-tolerant elastic training framework for PyTorch
1minitron. A mini Megatron 3D parallelism library for FSDL blog
1exex. Experiment codebase for chemchem package
1paulinet. Model the wave function ansatz using a neural network
1lexios. Symbolic framework for scientific calculation
1dreamerv3. Implementation of DreamerV3: Mastering Diverse Domains through World Models
1