hkproj / pytorch-transformer-distributed
Distributed training (multi-node) of a Transformer model
☆66Updated 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
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆46Updated 11 months ago
- ☆163Updated 4 months ago
- minimal GRPO implementation from scratch☆88Updated 2 months ago
- Notes and commented code for RLHF (PPO)☆90Updated last year
- LoRA and DoRA from Scratch Implementations☆202Updated last year
- ☆85Updated 7 months ago
- ☆186Updated 3 months ago
- GPU Kernels☆174Updated 2 weeks ago
- Notes about LLaMA 2 model☆59Updated last year
- An extension of the nanoGPT repository for training small MOE models.☆140Updated 2 months ago
- RL significantly the reasoning capability of Qwen2.5-1.5B-Instruct☆28Updated 2 months ago
- LLaMA 2 implemented from scratch in PyTorch☆324Updated last year
- Notes on quantization in neural networks☆82Updated last year
- ☆155Updated last year
- Prune transformer layers☆69Updated 11 months ago
- Fine-tune an LLM to perform batch inference and online serving.☆110Updated this week
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆294Updated last week
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆44Updated 8 months ago
- Building GPT ...☆17Updated 5 months ago
- ☆30Updated last week
- Student version of Assignment 1 for Stanford CS336 - Language Modeling From Scratch☆67Updated 3 weeks ago
- A set of scripts and notebooks on LLM finetunning and dataset creation☆109Updated 7 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆180Updated this week
- "LLM from Zero to Hero: An End-to-End Large Language Model Journey from Data to Application!"☆27Updated 2 weeks ago
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆102Updated last year
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆103Updated last week
- Reference implementation of Mistral AI 7B v0.1 model.☆28Updated last year
- Advanced NLP, Spring 2025 https://cmu-l3.github.io/anlp-spring2025/☆51Updated last month
- ☆42Updated last year