hkproj / pytorch-transformer-distributedLinks
Distributed training (multi-node) of a Transformer model
☆90Updated 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:
- An extension of the nanoGPT repository for training small MOE models.☆219Updated 9 months ago
- LLaMA 2 implemented from scratch in PyTorch☆363Updated 2 years ago
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆118Updated 2 years ago
- minimal GRPO implementation from scratch☆100Updated 9 months ago
- Notes and commented code for RLHF (PPO)☆120Updated last year
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆49Updated last year
- ☆45Updated 7 months ago
- ☆228Updated 11 months ago
- ☆99Updated last year
- Notes on Direct Preference Optimization☆23Updated last year
- ☆225Updated last month
- GPU Kernels☆212Updated 8 months ago
- LoRA and DoRA from Scratch Implementations☆214Updated last year
- Prune transformer layers☆74Updated last year
- RL significantly the reasoning capability of Qwen2.5-1.5B-Instruct☆31Updated 10 months ago
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆74Updated 2 years ago
- making the official triton tutorials actually comprehensible☆80Updated 4 months ago
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆138Updated 7 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆195Updated 6 months ago
- Notes about LLaMA 2 model☆71Updated 2 years ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆327Updated last month
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆351Updated 7 months ago
- ☆178Updated last year
- From scratch implementation of a vision language model in pure PyTorch☆254Updated last year
- ☆45Updated 7 months ago
- Advanced NLP, Spring 2025 https://cmu-l3.github.io/anlp-spring2025/☆69Updated 9 months ago
- A set of scripts and notebooks on LLM finetunning and dataset creation☆112Updated last year
- Notes on quantization in neural networks☆114Updated 2 years ago
- The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".☆186Updated last month
- Survey: A collection of AWESOME papers and resources on the latest research in Mixture of Experts.☆139Updated last year