hkproj / pytorch-transformer-distributedLinks
Distributed training (multi-node) of a Transformer model
☆83Updated 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☆21Updated last year
- LLaMA 2 implemented from scratch in PyTorch☆350Updated last year
- minimal GRPO implementation from scratch☆96Updated 6 months ago
- ☆94Updated 11 months ago
- LoRA and DoRA from Scratch Implementations☆211Updated last year
- ☆199Updated 8 months ago
- Notes and commented code for RLHF (PPO)☆106Updated last year
- LORA: Low-Rank Adaptation of Large Language Models implemented using PyTorch☆116Updated 2 years ago
- An extension of the nanoGPT repository for training small MOE models.☆185Updated 6 months ago
- Complete implementation of Llama2 with/without KV cache & inference 🚀☆48Updated last year
- RL significantly the reasoning capability of Qwen2.5-1.5B-Instruct☆30Updated 6 months ago
- Notes on quantization in neural networks☆98Updated last year
- Notes about LLaMA 2 model☆68Updated 2 years ago
- Prune transformer layers☆69Updated last year
- GPU Kernels☆193Updated 4 months ago
- A set of scripts and notebooks on LLM finetunning and dataset creation☆110Updated 11 months ago
- The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".☆175Updated 5 months ago
- ☆44Updated 3 months ago
- A simplified implementation for experimenting with RLVR on GSM8K, This repository provides a starting point for exploring reasoning.☆122Updated 7 months ago
- Unofficial implementation of https://arxiv.org/pdf/2407.14679☆48Updated last year
- ☆216Updated 7 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆269Updated last month
- This repository contains an implementation of the LLaMA 2 (Large Language Model Meta AI) model, a Generative Pretrained Transformer (GPT)…☆71Updated last year
- nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)☆118Updated 4 months ago
- From scratch implementation of a vision language model in pure PyTorch☆239Updated last year
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆334Updated 4 months ago
- Notes about "Attention is all you need" video (https://www.youtube.com/watch?v=bCz4OMemCcA)☆308Updated 2 years ago
- A easy, reliable, fluid template for python packages complete with docs, testing suites, readme's, github workflows, linting and much muc…☆188Updated last week
- A framework to study AI models in Reasoning, Alignment, and use of Memory (RAM).☆284Updated last week
- ☆44Updated 4 months ago