kklemon / FlashPerceiverLinks
Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
☆26Updated 10 months ago
Alternatives and similar repositories for FlashPerceiver
Users that are interested in FlashPerceiver are comparing it to the libraries listed below
Sorting:
- Flash Attention Triton kernel with support for second-order derivatives☆75Updated last week
- ☆53Updated last year
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆107Updated 11 months ago
- ☆32Updated last year
- σ-GPT: A New Approach to Autoregressive Models☆67Updated last year
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆56Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- ☆34Updated last year
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆129Updated 9 months ago
- A simple, performant and scalable JAX-based world modeling codebase☆73Updated this week
- Implementation of the proposed Spline-Based Transformer from Disney Research☆103Updated 10 months ago
- Implementation of a framework for Genie2 in Pytorch☆151Updated 8 months ago
- Focused on fast experimentation and simplicity☆75Updated 8 months ago
- ☆34Updated last year
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- ☆70Updated last year
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆90Updated 3 weeks ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 7 months ago
- ☆25Updated 3 months ago
- ☆34Updated 9 months ago
- RS-IMLE☆42Updated 9 months ago
- Code for the paper "Function-Space Learning Rates"☆23Updated 3 months ago
- ☆82Updated last year
- Exploration into the Firefly algorithm in Pytorch☆40Updated 7 months ago
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆67Updated 3 months ago
- H-Net Dynamic Hierarchical Architecture☆79Updated this week
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆53Updated 9 months ago
- ☆33Updated 8 months ago
- ☆23Updated last year