google-deepmind / alta
☆22Updated 6 months ago
Alternatives and similar repositories for alta:
Users that are interested in alta are comparing it to the libraries listed below
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆26Updated 11 months ago
- Using FlexAttention to compute attention with different masking patterns☆43Updated 6 months ago
- [ICLR 2025] "Training LMs on Synthetic Edit Sequences Improves Code Synthesis" (Piterbarg, Pinto, Fergus)☆18Updated 2 months ago
- ☆53Updated last year
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated 10 months ago
- ☆77Updated 7 months ago
- Stick-breaking attention☆50Updated last month
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 4 months ago
- Learning from preferences is a common paradigm for fine-tuning language models. Yet, many algorithmic design decisions come into play. Ou…☆28Updated 11 months ago
- Efficient Scaling laws and collaborative pretraining.☆16Updated 2 months ago
- ☆31Updated last year
- ☆31Updated last year
- GoldFinch and other hybrid transformer components☆45Updated 8 months ago
- ☆19Updated 10 months ago
- Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind☆123Updated 7 months ago
- Minimum Description Length probing for neural network representations☆19Updated 2 months ago
- Universal Neurons in GPT2 Language Models☆27Updated 10 months ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- ☆45Updated last year
- ☆31Updated 3 months ago
- Code for the paper "Function-Space Learning Rates"☆18Updated this week
- ☆60Updated 11 months ago
- ☆51Updated 10 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆97Updated 6 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 4 months ago
- Implementation of Direct Preference Optimization☆15Updated last year
- Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers☆17Updated last month
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆71Updated 5 months ago
- Memory Mosaics are networks of associative memories working in concert to achieve a prediction task.☆40Updated 2 months ago
- ☆76Updated 9 months ago