lucidrains / flash-attentionView external linksLinks
Fast and memory-efficient exact attention
☆20Jul 22, 2024Updated last year
Alternatives and similar repositories for flash-attention
Users that are interested in flash-attention are comparing it to the libraries listed below
Sorting:
- 🎓 The chrome extension to make learning from YouTube faster & easier.☆11Jan 9, 2022Updated 4 years ago
- ☆15Jan 27, 2025Updated last year
- ☆13Apr 16, 2022Updated 3 years ago
- tensorflow2 implementation of SnapMix as described in SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data☆11Feb 4, 2021Updated 5 years ago
- A real-time collaborative diagram editor and document library that enables users to chat, create, & communicate visually in real-time. (C…☆19Feb 3, 2026Updated last week
- FAQ for University of CaliforniaSanta Cruz 2019 Incoming Grads☆11Apr 4, 2019Updated 6 years ago
- ☆10May 24, 2020Updated 5 years ago
- Code for H. Narasimhan, "Learning with Complex Loss Functions and Constraints", AISTATS 2018☆11Mar 21, 2018Updated 7 years ago
- ☆10Dec 12, 2023Updated 2 years ago
- Radam+lookahead implemented by tensorflow☆11Oct 14, 2019Updated 6 years ago
- ☆12Jun 12, 2017Updated 8 years ago
- The package used to win the ARIEL challenge 2023 by the AstroAI team @ CfA | Harvard & Smithsonian☆11Jul 31, 2023Updated 2 years ago
- Pixie pipeline described in Liu et al., Robust phenotyping of highly multiplexed tissue imaging data using pixel-level clustering☆12Aug 3, 2023Updated 2 years ago
- ☆11Oct 28, 2023Updated 2 years ago
- This project implements the Titans architecture from the paper "Titans: Learning to Memorize at Test Time" for market data prediction.☆11Jan 19, 2025Updated last year
- ☆11Mar 3, 2025Updated 11 months ago
- FocusFlow: Boosting Key-Points Optical Flow Estimation for Autonomous Driving☆10Jan 22, 2024Updated 2 years ago
- A PyTorch implementation of [VCT](https://github.com/google-research/google-research/tree/master/vct)☆10Nov 25, 2022Updated 3 years ago
- ☆11Mar 28, 2021Updated 4 years ago
- ☆15Oct 20, 2025Updated 3 months ago
- Implementation of Cross Transformer for spatially-aware few-shot transfer, in Pytorch☆54Mar 30, 2021Updated 4 years ago
- Ultra-minimal autoregressive diffusion model for image generation☆21Dec 26, 2025Updated last month
- 9th solution☆11Oct 11, 2022Updated 3 years ago
- NoC based MPSoC☆11Jul 17, 2014Updated 11 years ago
- This is the official code for the ACL 2025 paper "GRAM: Generative Recommendation via Semantic-aware Multi-granular Late Fusion".☆27Aug 30, 2025Updated 5 months ago
- Experimental RISC-V assembler code snippets☆10Oct 23, 2019Updated 6 years ago
- Neptune - TensorBoard integration 🧩 Experiment tracking with advanced UI, collaborative features, and user access management.☆13Sep 4, 2025Updated 5 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- FU-net: Multi-class Image Segmentation using Feedback Weighted U-net☆12Apr 3, 2023Updated 2 years ago
- Pytorch implementation of "Very Deep Graph Neural Networks via Noise Regularisation"☆10Aug 22, 2021Updated 4 years ago
- 模式识别期末项目-基于Keras的人物面部表情识别☆11Jun 25, 2019Updated 6 years ago
- ZET-Speech: Zero-shot adaptive Emotion-controllable Text-to-Speech Synthesis with Diffusion and Style-based Models (TTS)☆10Mar 9, 2024Updated last year
- Dynamic Time-Aware Attention to Speaker Roles and Contexts for Spoken Language Understanding☆14Sep 28, 2017Updated 8 years ago
- Provably Secure Steganography☆14Sep 13, 2025Updated 5 months ago
- Release code for light-weight calibrator: a separable component for unsupervised domain adaptation☆13Jul 17, 2021Updated 4 years ago
- 🔮Getting Started with Pytorch: Text Classification Tutorial☆15Oct 25, 2017Updated 8 years ago
- Deep Learning Models for AKI Prediction on MIMIC-III and eICU database☆11Nov 7, 2019Updated 6 years ago
- Implementation of "Denoise Pretraining on Non-equilibrium Molecular Conformations for Accurate and Transferable Neural Potentials" in PyT…☆14Jul 26, 2023Updated 2 years ago
- Official implementation of "MadCLIP: Few-shot Medical Anomaly Detection with CLIP" (MICCAI 2025, Early Accepted).☆25Jul 24, 2025Updated 6 months ago