attentionmech / tensorlensLinks
aesthetic tensor visualiser
☆26Updated 5 months ago
Alternatives and similar repositories for tensorlens
Users that are interested in tensorlens are comparing it to the libraries listed below
Sorting:
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆105Updated 7 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆83Updated last month
- look how they massacred my boy☆63Updated 11 months ago
- A collection of lightweight interpretability scripts to understand how LLMs think☆56Updated last week
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Updated 5 months ago
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated 11 months ago
- An introduction to LLM Sampling☆79Updated 9 months ago
- ☆28Updated last year
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆55Updated 8 months ago
- ☆59Updated 7 months ago
- Plotting (entropy, varentropy) for small LMs☆98Updated 4 months ago
- alternative way to calculating self attention☆18Updated last year
- ☆40Updated last year
- Simple GRPO scripts and configurations.☆59Updated 8 months ago
- Repository to create traveling waves integrate special information through time☆55Updated last month
- Entropy Based Sampling and Parallel CoT Decoding☆17Updated 11 months ago
- explore token trajectory trees on instruct and base models☆133Updated 4 months ago
- Lego for GRPO☆29Updated 4 months ago
- smolLM with Entropix sampler on pytorch☆150Updated 11 months ago
- train entropix like a champ!☆20Updated 11 months ago
- Collection of autoregressive model implementation☆86Updated 5 months ago
- Port of Andrej Karpathy's nanoGPT to Apple MLX framework.☆112Updated last year
- ☆46Updated 6 months ago
- Simple Transformer in Jax☆139Updated last year
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆72Updated 7 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated 11 months ago
- Compiling useful links, papers, benchmarks, ideas, etc.☆45Updated 6 months ago
- Simple repository for training small reasoning models☆40Updated 8 months ago
- working implimention of deepseek MLA☆44Updated 8 months ago
- An example implementation of RLHF (or, more accurately, RLAIF) built on MLX and HuggingFace.☆32Updated last year