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:
- look how they massacred my boy☆64Updated 10 months ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆71Updated 4 months ago
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆172Updated 7 months ago
- smolLM with Entropix sampler on pytorch☆150Updated 10 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆55Updated 7 months ago
- ☆134Updated last year
- DeMo: Decoupled Momentum Optimization☆190Updated 9 months ago
- Train your own SOTA deductive reasoning model☆106Updated 6 months ago
- Collection of autoregressive model implementation☆86Updated 4 months ago
- Lego for GRPO☆29Updated 3 months ago
- EvaByte: Efficient Byte-level Language Models at Scale☆109Updated 4 months ago
- An introduction to LLM Sampling☆79Updated 8 months ago
- Plotting (entropy, varentropy) for small LMs☆98Updated 3 months ago
- Official repo for Learning to Reason for Long-Form Story Generation☆69Updated 4 months ago
- Simple & Scalable Pretraining for Neural Architecture Research☆291Updated 3 weeks ago
- rl from zero pretrain, can it be done? yes.☆265Updated 3 weeks ago
- NanoGPT (124M) quality in 2.67B tokens☆28Updated last month
- PyTorch implementation of models from the Zamba2 series.☆184Updated 7 months ago
- Simple Transformer in Jax☆139Updated last year
- Exploring Applications of GRPO☆247Updated 3 weeks ago
- Storing long contexts in tiny caches with self-study☆179Updated last week
- Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…☆64Updated 10 months ago
- working implimention of deepseek MLA☆44Updated 8 months ago
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated 10 months ago
- SIMD quantization kernels☆86Updated last week
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆83Updated 3 weeks ago
- Lightweight toolkit package to train and fine-tune 1.58bit Language models☆88Updated 3 months ago
- Entropy Based Sampling and Parallel CoT Decoding☆17Updated 11 months ago
- ☆119Updated 8 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Updated last month