hkproj / pytorch-transformer-distributedLinks
Distributed training (multi-node) of a Transformer model
☆72Updated last year
Alternatives and similar repositories for pytorch-transformer-distributed
Users that are interested in pytorch-transformer-distributed are comparing it to the libraries listed below
Sorting:
- Notes on Direct Preference Optimization☆19Updated last year
- ☆174Updated 5 months ago
- minimal GRPO implementation from scratch☆90Updated 3 months ago
- Advanced NLP, Spring 2025 https://cmu-l3.github.io/anlp-spring2025/☆55Updated 2 months ago
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆109Updated last year
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆47Updated last year
- ☆39Updated last month
- Notes and commented code for RLHF (PPO)☆96Updated last year
- RL significantly the reasoning capability of Qwen2.5-1.5B-Instruct☆29Updated 4 months ago
- ☆159Updated last year
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated last year
- ☆89Updated 9 months ago
- LoRA and DoRA from Scratch Implementations☆204Updated last year
- LLaMA 2 implemented from scratch in PyTorch☆335Updated last year
- GPU Kernels☆182Updated last month
- Notes about LLaMA 2 model☆61Updated last year
- ☆41Updated last month
- Prune transformer layers☆69Updated last year
- An extension of the nanoGPT repository for training small MOE models.☆152Updated 3 months ago
- "LLM from Zero to Hero: An End-to-End Large Language Model Journey from Data to Application!"☆30Updated this week
- A set of scripts and notebooks on LLM finetunning and dataset creation☆111Updated 8 months ago
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆68Updated last year
- ☆193Updated 4 months ago
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆105Updated last month
- From scratch implementation of a vision language model in pure PyTorch☆222Updated last year
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆185Updated 3 weeks ago
- Notes on quantization in neural networks☆86Updated last year
- making the official triton tutorials actually comprehensible☆41Updated 3 months ago
- Student version of Assignment 1 for Stanford CS336 - Language Modeling From Scratch☆174Updated 2 months ago
- ☆14Updated 2 months ago