PrototypeCommon
nebius/kvax
A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
- Owner
- nebius (organization)
- Stars
- Stars: 177
- Forks
- Forks: 11
- Language
- Python
- License
- Apache-2.0
- Created
- 2025
- Tier
- Prototype (100 to 249 ★) What this tier means
More repos by nebius
All 4 repos by nebius- nebius/soperatorStars: 456GoAlpha
- nebius/token-factory-cookbookStars: 315Jupyter NotebookAlpha
- nebius/nebius-solutions-libraryStars: 123HCLPrototype
Nearby in Python
All Python repos- yrui-cmd/cell_su7Stars: 177PythonPrototype
- Inferact/tpu-megakernelsStars: 177PythonPrototype
- duyongan/dots-agentStars: 177PythonPrototype
- pyudev/pyudevStars: 177PythonPrototype
- mhl/guardian-for-kindleStars: 177PythonPrototype
- mattharrison/Git-Supervisual-CheatsheetStars: 177PythonPrototype