KindXiaoming / physics_of_skill_learningLinks
We study toy models of skill learning.
☆29Updated 6 months ago
Alternatives and similar repositories for physics_of_skill_learning
Users that are interested in physics_of_skill_learning are comparing it to the libraries listed below
Sorting:
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆81Updated 9 months ago
- ☆83Updated 11 months ago
- Implementation of Infini-Transformer in Pytorch☆110Updated 7 months ago
- A repository for research on medium sized language models.☆78Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆120Updated 9 months ago
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- ☆81Updated last year
- [ICLR 2025] Monet: Mixture of Monosemantic Experts for Transformers☆70Updated last month
- Official Code Repository for the paper "Key-value memory in the brain"☆27Updated 5 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 7 months ago
- Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…☆53Updated last year
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- GoldFinch and other hybrid transformer components☆46Updated last year
- [ICLR'25] Code for KaSA, an official implementation of "KaSA: Knowledge-Aware Singular-Value Adaptation of Large Language Models"☆19Updated 6 months ago
- Implementations of attention with the softpick function, naive and FlashAttention-2☆81Updated 3 months ago
- Using FlexAttention to compute attention with different masking patterns☆44Updated 10 months ago
- Anchored Preference Optimization and Contrastive Revisions: Addressing Underspecification in Alignment☆60Updated 11 months ago
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆41Updated 9 months ago
- ☆48Updated 11 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated 11 months ago
- Minimum Description Length probing for neural network representations☆18Updated 6 months ago
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆47Updated 3 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆55Updated 4 months ago
- Here we will test various linear attention designs.☆62Updated last year
- Code for reproducing our paper "Low Rank Adapting Models for Sparse Autoencoder Features"☆12Updated 4 months ago
- ☆31Updated last year
- Unofficial Implementation of Selective Attention Transformer☆17Updated 9 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- Code for the paper "Cottention: Linear Transformers With Cosine Attention"☆17Updated 9 months ago
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆29Updated 3 months ago