Workflow

Jax-flash-attn2

Latest version: v0.0.3

The latest version of jax-flash-attn2 with no known security vulnerabilities is 0.0.3. We recommend installing version 0.0.3.

The information on this page was curated by experts in our Cybersecurity Intelligence Team.

Latest release
v0.0.3 at March 4, 2025
License
Apache-2.0 (Apache License 2.0)

Description

Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attention with support for different backends (GPU, TPU, CPU) and platforms (Triton, Pallas, JAX).

Resources

Vulnerabilities

See all vulnerabilities

No known vulnerabilities found

Versions (2)

See all versions

Has known vulnerabilities

  • 0.0.3
  • 0.0.1