PyTorch implementation of the Flash Spectral Transform Unit.
☆22Sep 19, 2024Updated last year
Alternatives and similar repositories for flash-stu
Users that are interested in flash-stu are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- ☆35Apr 12, 2024Updated 2 years ago
- ☆116Feb 25, 2025Updated last year
- ☆12Mar 7, 2022Updated 4 years ago
- image retrieval/tagging with CLIP☆13Jul 13, 2024Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- ☆13Jan 7, 2025Updated last year
- ☆33Oct 4, 2024Updated last year
- ☆22May 5, 2025Updated last year
- Training and evaluation code for the paper "Headless Language Models: Learning without Predicting with Contrastive Weight Tying" (https:/…☆29Apr 17, 2024Updated 2 years ago
- ☆13Jun 18, 2024Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19May 8, 2025Updated last year
- Code and data for paper "(How) do Language Models Track State?"☆23Mar 31, 2025Updated last year
- 全卷积网络进行语音降噪☆18Dec 8, 2021Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 方便扩展的Cuda算子理解和优化框架，仅用在学习使用☆18Jun 13, 2024Updated last year
- Notes for CIS 700 (Fall '19) at Syracuse U.☆13Nov 6, 2019Updated 6 years ago
- study of cutlass☆22Nov 10, 2024Updated last year
- ☆20Dec 24, 2024Updated last year
- ☆40Dec 14, 2025Updated 5 months ago
- Full spectrum sheaf neural network over arbitrary CW complexes.☆17Apr 1, 2026Updated 2 months ago
- It's a baby compiler. (Lean btw.)☆16May 19, 2025Updated last year
- A GPU FP32 computation method with Tensor Cores.☆27Dec 8, 2025Updated 6 months ago
- Accelerated First Order Parallel Associative Scan☆197Jan 7, 2026Updated 5 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆25Jun 6, 2024Updated 2 years ago
- SMT-LIB benchmarks for shape computations from deep learning models in PyTorch☆18Dec 21, 2022Updated 3 years ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated 10 months ago
- CUDA implementation of RTXX algorithm for multiplication of matrix and its transpose X^T X☆19Jun 9, 2025Updated 11 months ago
- Official code for UnICORNN (ICML 2021)☆27Oct 1, 2021Updated 4 years ago
- KernelBench v2: Can LLMs Write GPU Kernels? - Benchmark with Torch -> Triton (and more!) problems☆23Jul 4, 2025Updated 11 months ago
- Experiments on Multi-Head Latent Attention☆101Aug 19, 2024Updated last year
- Sample Codes using NVSHMEM on Multi-GPU☆30Jan 22, 2023Updated 3 years ago
- Official Code Repository for the paper "Key-value memory in the brain"☆32Feb 25, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- FlashRNN - Fast RNN Kernels with I/O Awareness☆182Oct 20, 2025Updated 7 months ago
- Implementation of paper - RepVGG-GELAN: ENHANCED GELAN WITH VGG-STYLE CONVNETS FOR BRAIN TUMOR DETECTION☆10Jul 19, 2025Updated 10 months ago
- DeeperGEMM: crazy optimized version☆86May 5, 2025Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆41Oct 23, 2025Updated 7 months ago
- ☆11Jul 26, 2024Updated last year
- Transformers components but in Triton☆34May 9, 2025Updated last year
- Awesome Triton Resources☆42Apr 27, 2025Updated last year