xjdr-alt / entropix-trainerLinks
train entropix like a champ!
☆20Updated last year
Alternatives and similar repositories for entropix-trainer
Users that are interested in entropix-trainer are comparing it to the libraries listed below
Sorting:
- smol models are fun too☆93Updated 11 months ago
- look how they massacred my boy☆63Updated last year
- Plotting (entropy, varentropy) for small LMs☆98Updated 5 months ago
- Entropy Based Sampling and Parallel CoT Decoding☆17Updated last year
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated 11 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆107Updated 7 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆72Updated 8 months ago
- smolLM with Entropix sampler on pytorch☆150Updated 11 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated 11 months ago
- NSA Triton Kernels written with GPT5 and Opus 4.1☆65Updated 2 months ago
- Synthetic data derived by templating, few shot prompting, transformations on public domain corpora, and monte carlo tree search.☆32Updated 2 weeks ago
- Approximating the joint distribution of language models via MCTS☆22Updated 11 months ago
- ☆68Updated 5 months ago
- ☆40Updated last year
- Compiling useful links, papers, benchmarks, ideas, etc.☆45Updated 7 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆84Updated 2 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆58Updated last week
- SIMD quantization kernels☆87Updated last month
- Train your own SOTA deductive reasoning model☆108Updated 7 months ago
- Simple Transformer in Jax☆139Updated last year
- ☆124Updated 10 months ago
- Lego for GRPO☆30Updated 4 months ago
- ☆21Updated 9 months ago
- A graph visualization of attention☆57Updated 5 months ago
- DeMo: Decoupled Momentum Optimization☆194Updated 10 months ago
- Storing long contexts in tiny caches with self-study☆201Updated last week
- ☆135Updated 7 months ago
- An introduction to LLM Sampling☆79Updated 10 months ago
- Modded vLLM to run pipeline parallelism over public networks☆39Updated 5 months ago
- ☆136Updated last year