vulkomilev

@fchollet I have made some changes

New commit

I think that test are passing now @fchollet

yes I will do it today

@fchollet I think that I have scan test in CoreOpsStaticShapeTest

@gbaned done

@gbaned done

I have used /keras/src/layers/attention/ directory as a template for implementing a flash attention but I don't understand how the mask is generated in the benchmark. I need one but I...

forgot to add that I have changed the git pull to use ssh

I will try to work on it