Skip to content

Add fast_softmax parameter to _gradient_checkpointing_func

e1cd8e0
Select commit
Loading
Failed to load commit list.
Merged

Enable fast softmax mode in FusedSDPA #159

Add fast_softmax parameter to _gradient_checkpointing_func
e1cd8e0
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs