JAX bindings for the FlashAttention 3 kernels
☆26Sep 9, 2026Updated last month
Alternatives and similar repositories for flash-attn3-jax
Users that are interested in flash-attn3-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- JAX bindings for Flash Attention v2☆110Feb 28, 2026Updated 8 months ago
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated 2 years ago
- 📖 Application développée pour simplifier l'analyse et la gestion des textes juridiques français et européens en utilisant des modèles d'…☆15Aug 20, 2025Updated last year
- This is the official implementation of the architecture presented in Leveraging Morphology for Historical Script Metrological Analysis (I…☆16Oct 14, 2026Updated 2 weeks ago
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)☆16Aug 6, 2024Updated 2 years ago
- JAX PJRT plugin for Apple Silicon built on XLA:GPU☆19Oct 11, 2026Updated 2 weeks ago
- Official implementation of the CVPR 2026 paper Illustrator's Depth☆25Jul 26, 2026Updated 3 months ago
- PyTorch implementation of Robust Subspace Recovery Layer for Unsupervised Anomaly Detection https://arxiv.org/abs/1904.00152☆15Apr 26, 2021Updated 5 years ago
- ☆11Sep 13, 2023Updated 3 years ago
- This Repository allows to super fast download historical ohlcv data from binance.☆12Nov 27, 2020Updated 5 years ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆156Oct 16, 2025Updated last year
- ☆46Aug 3, 2026Updated 2 months ago
- [CVPRW2025] Studying Image Diffusion Features for Zero-Shot Video Object Segmentation☆14Apr 11, 2025Updated last year
- 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.
- Official repository of the paper "Text region detection in historical astronomical diagrams" (ICDAR 2026).☆20Jun 23, 2026Updated 4 months ago
- ☆16Sep 24, 2026Updated last month
- Tools and Utils for Experiments (TUX)☆17Jan 28, 2025Updated last year
- Contains scripts for the PSI competition.☆11Dec 11, 2023Updated 2 years ago
- ☆18Mar 26, 2026Updated 7 months ago
- ☆26May 3, 2024Updated 2 years ago
- Historical Diagram Vectorization☆20Nov 25, 2025Updated 11 months ago
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated 2 years ago
- https://www.nodecafe.co☆33Apr 12, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Code for "What really matters in matrix-whitening optimizers?"☆25Oct 31, 2025Updated 11 months ago
- ☆23Aug 5, 2024Updated 2 years ago
- LLM pretraining in pure JAX☆16Apr 1, 2026Updated 6 months ago
- ☆23Mar 30, 2023Updated 3 years ago
- Implementation of the article Online Structured Laplace Approximations For Overcoming Catastrophic Forgetting (https://arxiv.org/pdf/1805…☆10Feb 1, 2019Updated 7 years ago
- Official code for the paper "BECAME: BayEsian Continual Learning with Adaptive Model MErging" (ICML 2025).☆12Jul 4, 2025Updated last year
- ☆13Feb 25, 2025Updated last year
- ☆21Oct 5, 2026Updated 3 weeks ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆537Updated this week
- 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.
- utility functions for CIL☆19Jun 18, 2024Updated 2 years ago
- Adds a tag indicating whether a user follows you on GitHub☆13Jan 18, 2024Updated 2 years ago
- ☆25Oct 6, 2025Updated last year
- Master Thesis Project in Computer Engineering at Aarhus University 2024 on "Simulating Multi-agent Path Planning in Complex environments …☆18Oct 12, 2025Updated last year
- ☆19Nov 23, 2022Updated 3 years ago
- ☆19Sep 25, 2026Updated last month
- ☆10Oct 8, 2026Updated 3 weeks ago