Silent-Zebra / twisted-smc-lmLinks
☆29Updated 2 months ago
Alternatives and similar repositories for twisted-smc-lm
Users that are interested in twisted-smc-lm are comparing it to the libraries listed below
Sorting:
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆59Updated 3 months ago
- ☆52Updated last year
- ☆53Updated 8 months ago
- ☆32Updated 7 months ago
- ☆19Updated 2 months ago
- Stick-breaking attention☆56Updated 2 months ago
- ☆37Updated last year
- Universal Neurons in GPT2 Language Models☆29Updated last year
- ☆85Updated last year
- ☆51Updated last year
- Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.☆36Updated last year
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆73Updated 7 months ago
- ☆23Updated 4 months ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 7 months ago
- Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…☆16Updated 6 months ago
- Experiments on the impact of depth in transformers and SSMs.☆30Updated 6 months ago
- The Energy Transformer block, in JAX☆56Updated last year
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Code for "Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining"☆16Updated last month
- Reinforcement Learning via Regressing Relative Rewards☆32Updated 5 months ago
- ☆65Updated 11 months ago
- ☆26Updated last year
- ☆32Updated last year
- ☆19Updated 10 months ago
- Official implementation of the transformer (TF) architecture suggested in a paper entitled "Looped Transformers as Programmable Computers…☆27Updated 2 years ago
- ☆9Updated 2 years ago
- Sparse Autoencoder Training Library☆50Updated last month
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆81Updated last year
- ☆45Updated last year
- Code associated to papers on superposition (in ML interpretability)☆28Updated 2 years ago