JoeLi12345 / nGPTLinks
an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)
☆105Updated 6 months ago
Alternatives and similar repositories for nGPT
Users that are interested in nGPT are comparing it to the libraries listed below
Sorting:
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Updated 5 months ago
- look how they massacred my boy☆63Updated 11 months ago
- ☆135Updated last year
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆172Updated 8 months ago
- Collection of autoregressive model implementation☆86Updated 5 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆55Updated 8 months ago
- Train your own SOTA deductive reasoning model☆107Updated 6 months ago
- smolLM with Entropix sampler on pytorch☆150Updated 11 months ago
- DeMo: Decoupled Momentum Optimization☆193Updated 10 months ago
- Lego for GRPO☆29Updated 4 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆83Updated last month
- Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…☆65Updated this week
- An introduction to LLM Sampling☆79Updated 9 months ago
- PyTorch implementation of models from the Zamba2 series.☆185Updated 8 months ago
- working implimention of deepseek MLA☆44Updated 8 months ago
- An easy-to-understand framework for LLM samplers that rewind and revise generated tokens☆146Updated 7 months ago
- NanoGPT (124M) quality in 2.67B tokens☆28Updated 2 weeks ago
- ☆123Updated 9 months ago
- Simple & Scalable Pretraining for Neural Architecture Research☆296Updated last month
- Plotting (entropy, varentropy) for small LMs☆99Updated 4 months ago
- Cerule - A Tiny Mighty Vision Model☆68Updated last year
- EvaByte: Efficient Byte-level Language Models at Scale☆109Updated 5 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Updated 2 months ago
- Lightweight toolkit package to train and fine-tune 1.58bit Language models☆89Updated 4 months ago
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆201Updated last year
- Exploring Applications of GRPO☆250Updated last month
- RWKV-7: Surpassing GPT☆96Updated 10 months ago
- Simple Transformer in Jax☆139Updated last year
- Official repo for Learning to Reason for Long-Form Story Generation☆72Updated 5 months ago
- rl from zero pretrain, can it be done? yes.☆274Updated last week