rachtsy / KPCA_codeLinks
Implementation for robust ViT and scaled attention
☆20Updated 6 months ago
Alternatives and similar repositories for KPCA_code
Users that are interested in KPCA_code are comparing it to the libraries listed below
Sorting:
- Minimum Description Length probing for neural network representations☆20Updated 9 months ago
- Fork of Flame repo for training of some new stuff in development☆18Updated 2 weeks ago
- 📄Small Batch Size Training for Language Models☆63Updated 3 weeks ago
- ☆86Updated last year
- Code and pretrained models for the paper: "MatMamba: A Matryoshka State Space Model"☆61Updated 11 months ago
- ☆58Updated last year
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- ☆13Updated 7 months ago
- ☆81Updated last year
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆147Updated 3 weeks ago
- Code for the paper "Function-Space Learning Rates"☆23Updated 4 months ago
- Anchored Preference Optimization and Contrastive Revisions: Addressing Underspecification in Alignment☆60Updated last year
- Collection of autoregressive model implementation☆86Updated 6 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆84Updated 11 months ago
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated last year
- H-Net Dynamic Hierarchical Architecture☆80Updated last month
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 5 months ago
- JAX Scalify: end-to-end scaled arithmetics☆16Updated 11 months ago
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆72Updated last week
- Jax like function transformation engine but micro, microjax☆33Updated last year
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆72Updated 5 months ago
- Implementation of Mind Evolution, Evolving Deeper LLM Thinking, from Deepmind☆57Updated 4 months ago
- ☆22Updated 10 months ago
- Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise☆39Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆102Updated 10 months ago
- Using FlexAttention to compute attention with different masking patterns☆47Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Updated 3 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 4 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆29Updated 8 months ago