JAX bindings for the FlashAttention 3 kernels
☆25Mar 9, 2026Updated 5 months ago
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:
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 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 11 months ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆17Aug 1, 2022Updated 4 years ago
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago
- Plotutils+ is a fork of the GNU plotutils package, with fixes and extensions☆13May 14, 2024Updated 2 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
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆152Oct 16, 2025Updated 9 months ago
- ☆24Feb 17, 2020Updated 6 years ago
- Historical Diagram Vectorization☆20Nov 25, 2025Updated 8 months ago
- https://www.nodecafe.co☆34Apr 12, 2024Updated 2 years ago
- ☆17Mar 26, 2026Updated 4 months ago
- Contains scripts for the PSI competition.☆11Dec 11, 2023Updated 2 years ago
- Full Bayesian inference for neural networks using TensorFlow☆15Aug 19, 2021Updated 4 years ago
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆11Nov 23, 2023Updated 2 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.
- ☆27May 3, 2024Updated 2 years ago
- ☆23Aug 5, 2024Updated 2 years ago
- A real-time and multilingual speech translation model☆268Feb 13, 2026Updated 5 months ago
- ☆11Dec 23, 2023Updated 2 years ago
- Code for "What really matters in matrix-whitening optimizers?"☆25Oct 31, 2025Updated 9 months ago
- LLM pretraining in pure JAX☆16Apr 1, 2026Updated 4 months ago
- Memory efficient one-hot array encodings☆19Apr 17, 2026Updated 3 months ago
- utility functions for CIL☆19Jun 18, 2024Updated 2 years ago
- Explorations into the proposed NeuralGrok from Zhou et al. of EPFL☆15Oct 18, 2025Updated 9 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.
- ☆30Apr 15, 2024Updated 2 years ago
- ☆19Nov 23, 2022Updated 3 years ago
- PysynthMRI is a python open-source tool which provides a user-friendly interface to synthesize contrast weighted images.☆27Nov 10, 2025Updated 9 months ago
- SeismicNet: Physics-informed neural networks for seismic wave modeling in semi-infinite domain☆22May 5, 2024Updated 2 years ago
- qwen3 experiments☆33Jul 1, 2025Updated last year
- ☆15Jan 23, 2025Updated last year
- [WACV 2024] MACP: Efficient Model Adaptation for Cooperative Perception.☆19May 3, 2024Updated 2 years ago
- Unofficial implementation of 'Large-Scale Text-to-Image Model with Inpainting is a Zero-Shot Subject-Driven Image Generator'☆10Dec 10, 2024Updated last year
- Orbax provides common checkpointing and persistence utilities for JAX users☆527Updated this week
- 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.
- The Newton-Muon optimizer☆29Jun 5, 2026Updated 2 months ago
- ☆11Oct 5, 2023Updated 2 years ago
- DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core compone…☆43Jan 11, 2026Updated 6 months ago
- Cavs: An Efficient Runtime System for Dynamic Neural Networks☆15Sep 18, 2020Updated 5 years ago
- Meetup Call for Paper pour soumettre vos idées de sujets☆11Nov 1, 2017Updated 8 years ago
- CLI utility to inspect and explore .safetensors and .gguf files☆62Jun 20, 2026Updated last month
- ☆56Jul 16, 2025Updated last year