Phylliida / MambaLensLinks
Mamba support for transformer lens
☆17Updated 10 months ago
Alternatives and similar repositories for MambaLens
Users that are interested in MambaLens are comparing it to the libraries listed below
Sorting:
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆75Updated 8 months ago
- Stick-breaking attention☆58Updated 2 weeks ago
- Code for reproducing our paper "Not All Language Model Features Are Linear"☆77Updated 7 months ago
- ☆53Updated last year
- ☆66Updated last year
- ☆82Updated 10 months ago
- Universal Neurons in GPT2 Language Models☆30Updated last year
- Simple and efficient pytorch-native transformer training and inference (batched)☆77Updated last year
- Sparse Autoencoder Training Library☆53Updated 2 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆78Updated last month
- ☆97Updated 9 months ago
- ☆48Updated last year
- Language models scale reliably with over-training and on downstream tasks☆97Updated last year
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- ☆11Updated 4 months ago
- The official implementation for Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free☆45Updated 2 months ago
- Replicating O1 inference-time scaling laws☆89Updated 7 months ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆19Updated 11 months ago
- ☆33Updated 6 months ago
- Open source replication of Anthropic's Crosscoders for Model Diffing☆57Updated 8 months ago
- ☆37Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆32Updated 8 months ago
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆33Updated 3 months ago
- ☆20Updated last year
- ☆81Updated last year
- ☆87Updated 11 months ago
- Investigating the generalization behavior of LM probes trained to predict truth labels: (1) from one annotator to another, and (2) from e…☆28Updated last year
- Delphi was the home of a temple to Phoebus Apollo, which famously had the inscription, 'Know Thyself.' This library lets language models …☆193Updated this week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆90Updated 3 weeks ago
- [NeurIPS 2024 Spotlight] Code and data for the paper "Finding Transformer Circuits with Edge Pruning".☆59Updated this week