shreyansh26 / An-Empirical-Model-of-Large-Batch-TrainingLinks
An approximate implementation of the OpenAI paper - An Empirical Model of Large-Batch Training for MNIST
☆11Updated 2 years ago
Alternatives and similar repositories for An-Empirical-Model-of-Large-Batch-Training
Users that are interested in An-Empirical-Model-of-Large-Batch-Training are comparing it to the libraries listed below
Sorting:
- ☆53Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆102Updated 2 weeks ago
- ☆86Updated last year
- Efficient PScan implementation in PyTorch☆16Updated last year
- Code and Configs for Asynchronous RLHF: Faster and More Efficient RL for Language Models☆64Updated 6 months ago
- RWKV-X is a Linear Complexity Hybrid Language Model based on the RWKV architecture, integrating Sparse Attention to improve the model's l…☆51Updated 3 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆89Updated last year
- Here we will test various linear attention designs.☆61Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆69Updated last year
- ☆41Updated 2 weeks ago
- Combining SOAP and MUON☆16Updated 8 months ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆33Updated last year
- [NeurIPS 2024] Low rank memory efficient optimizer without SVD☆30Updated 4 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Updated 7 months ago
- The official code of "Building on Efficient Foundations: Effectively Training LLMs with Structured Feedforward Layers"☆19Updated last year
- Muon fsdp 2☆44Updated 2 months ago
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆72Updated 2 weeks ago
- ☆48Updated last year
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆129Updated last year
- ☆56Updated last year
- Experimental playground for benchmarking language model (LM) architectures, layers, and tricks on smaller datasets. Designed for flexible…☆84Updated 2 weeks ago
- Custom triton kernels for training Karpathy's nanoGPT.☆19Updated last year
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆65Updated last year
- Some preliminary explorations of Mamba's context scaling.☆216Updated last year
- ☆45Updated last week
- ☆34Updated 7 months ago
- Experiments on the impact of depth in transformers and SSMs.☆36Updated last week
- Official implementation for Training LLMs with MXFP4☆101Updated 6 months ago
- ☆19Updated 5 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆162Updated 6 months ago