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:
- look how they massacred my boy☆63Updated last year
- Plotting (entropy, varentropy) for small LMs☆99Updated 8 months ago
- smol models are fun too☆93Updated last year
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆61Updated last year
- Entropy Based Sampling and Parallel CoT Decoding☆17Updated last year
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆110Updated 10 months ago
- ☆68Updated 8 months ago
- smolLM with Entropix sampler on pytorch☆149Updated last year
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆77Updated 11 months ago
- Official CLI and Python SDK for Prime Intellect - access GPU compute, remote sandboxes, RL environments, and distributed training infrast…☆148Updated this week
- ☆137Updated 10 months ago
- Train your own SOTA deductive reasoning model☆107Updated 10 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆59Updated 3 months ago
- Synthetic data derived by templating, few shot prompting, transformations on public domain corpora, and monte carlo tree search.☆32Updated 3 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆85Updated 5 months ago
- Lego for GRPO☆30Updated 8 months ago
- ☆40Updated last year
- Simple Transformer in Jax☆142Updated last year
- NSA Triton Kernels written with GPT5 and Opus 4.1☆70Updated 5 months ago
- ☆27Updated last year
- Compiling useful links, papers, benchmarks, ideas, etc.☆46Updated 10 months ago
- DeMo: Decoupled Momentum Optimization☆198Updated last year
- Modded vLLM to run pipeline parallelism over public networks☆40Updated 8 months ago
- An introduction to LLM Sampling☆79Updated last year
- Official repo for Learning to Reason for Long-Form Story Generation☆74Updated 9 months ago
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆175Updated last year
- SIMD quantization kernels☆94Updated 4 months ago
- Approximating the joint distribution of language models via MCTS☆22Updated last year
- explore token trajectory trees on instruct and base models☆150Updated 8 months ago