doomslide / attention-graphLinks
A graph visualization of attention
☆57Updated 5 months ago
Alternatives and similar repositories for attention-graph
Users that are interested in attention-graph are comparing it to the libraries listed below
Sorting:
- look how they massacred my boy☆63Updated last year
- Plotting (entropy, varentropy) for small LMs☆98Updated 5 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆49Updated last year
- smol models are fun too☆93Updated last year
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆62Updated last year
- Approximating the joint distribution of language models via MCTS☆22Updated last year
- Simple Transformer in Jax☆139Updated last year
- explore token trajectory trees on instruct and base models☆148Updated 5 months ago
- smolLM with Entropix sampler on pytorch☆150Updated last year
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆72Updated 9 months ago
- Synthetic data derived by templating, few shot prompting, transformations on public domain corpora, and monte carlo tree search.☆32Updated last month
- ☆68Updated 5 months ago
- ☆40Updated last year
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆84Updated 2 months ago
- ComplexTensor: Machine Learning By Bridging Classical and Quantum Computation☆77Updated 11 months ago
- Turing machines, Rule 110, and A::B reversal using Claude 3 Opus.☆58Updated last year
- An introduction to LLM Sampling☆79Updated 11 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆107Updated 8 months ago
- train entropix like a champ!☆20Updated last year
- ☆125Updated 10 months ago
- PageRank for LLMs☆51Updated 2 months ago
- Repository to create traveling waves integrate special information through time☆56Updated 3 months ago
- Official CLI and Python SDK for Prime Intellect - access GPU compute, remote sandboxes, RL environments, and distributed training infrast…☆106Updated this week
- Benchmark for LLMs playing full press diplomacy☆56Updated 8 months ago
- Gradient descent is cool and all, but what if we could delete it?☆104Updated 2 months ago
- j1-micro (1.7B) & j1-nano (600M) are absurdly tiny but mighty reward models.☆97Updated 3 months ago
- SIMD quantization kernels☆92Updated 2 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆58Updated 3 weeks ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆72Updated 6 months ago
- A strongly typed Python DSL for developing message passing multi agent systems☆53Updated last year