xla
Move requires_jax to inner flash_attention functions
#9098

Open

Move requires_jax to inner flash_attention functions #9098

bhavya01 wants to merge 2 commits into master from bhavya01-patch-23

Move requires_jax to inner flash_attention functions

50043515

run yapf

045c95d9

bhavya01 requested a review from

tengyifei 1 year ago

tengyifei requested changes on 2025-05-06

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone