flash-attention-jax

lucidrains

Implementation of Flash Attention in Jax

artificial-intelligence

View on GitHub

205 Stars

24 Forks

205 Watchers

Python Language

mit License

Cost to Build

$7.1K

Market Value

$16.7K

How is this calculated?

Growth over time

3 data points · 2022-11-01 → 2025-03-01

Stars Forks Watchers

💬

How do you feel about this project?

Ask AI about flash-attention-jax

Question copied to clipboard

What is the lucidrains/flash-attention-jax GitHub project? Description: "Implementation of Flash Attention in Jax". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone flash-attention-jax

Clone via HTTPS

git clone https://github.com/lucidrains/flash-attention-jax.git

Clone via SSH

[email protected]:lucidrains/flash-attention-jax.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the flash-attention-jax issue tracker:

Open GitHub Issues

Similar to flash-attention-jax

deeplearning4j machine-learning-for-software-engineers incubator-mxnet spaCy cheatsheets-ai gun php-ml TensorLayer awesome-artificial-intelligence AlgoWiki papers-I-read kong EmojiIntelligence PyGame-Learning-Environment deep-trading-agent caffe2 AirSim pipeline diffbot-php-client mycroft-core iOS_ML warriorjs nd4j incubator-kie-optaplanner ML-for-High-Schoolers Dragonfire auto_ml gophernotes deeplearning4j-examples DeepPavlov