zaydzuhri / softpick-attention
Implementations of attention with the softpick function, naive and FlashAttention-2
☆28Updated last week
Alternatives and similar repositories for softpick-attention:
Users that are interested in softpick-attention are comparing it to the libraries listed below
- Fork of Flame repo for training of some new stuff in development☆11Updated last week
- ☆33Updated 7 months ago
- ☆26Updated last year
- ☆22Updated 10 months ago
- Focused on fast experimentation and simplicity☆71Updated 4 months ago
- research impl of Native Sparse Attention (2502.11089)☆53Updated 2 months ago
- ☆78Updated 10 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆31Updated 6 months ago
- Code for the paper "Function-Space Learning Rates"☆19Updated 2 weeks ago
- ☆53Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆63Updated 2 weeks ago
- Exploration into the Firefly algorithm in Pytorch☆38Updated 2 months ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆50Updated last month
- supporting pytorch FSDP for optimizers☆80Updated 4 months ago
- ☆21Updated 4 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated last month
- ☆34Updated 4 months ago
- Latent Large Language Models☆18Updated 8 months ago
- NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…☆42Updated 5 months ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆68Updated 2 weeks ago
- ☆49Updated last year
- working implimention of deepseek MLA☆41Updated 3 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆101Updated 3 weeks ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆89Updated last month
- ☆31Updated last year
- ☆52Updated 5 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆24Updated 3 months ago
- This repo is based on https://github.com/jiaweizzhao/GaLore☆27Updated 7 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆49Updated last month