Factual package intelligence from PyPI

flash-attn-jax 0.6.2 package intelligence

Flash Attention port for JAX

pip install flash-attn-jax

Current version
0.6.2
Python requirement
>=3.11
License
BSD-3-Clause
Distribution type
Includes compiled wheel
Release files
5 (4 wheels, 1 source)
Download size
316.6 MiB
Release history
9 releases with files
Median release cadence
20 days

flash-attn-jax dependencies

PyPI declares 2 unique dependency rules for this release. Environment markers are shown when supplied by the project.

  1. nanobind >=2.0.0
  2. jax <0.9.0,>=0.5.0

Compatibility and release files

flash-attn-jax publishes 4 wheels and 1 source archive for version 0.6.2. Wheel platform tags: manylinux_2_24_x86_64.manylinux_2_28_x86_64.

No version-specific Python classifiers are declared.

Release activity and sources

PyPI lists 9 releases with files. The first dated release is ; 6 releases fall within the 365 days preceding the latest dated release. The current release files were uploaded on . The preceding dated release was .