What I meant was, depending on what you want to do, you might get by with scripting the cell operations instead of writing CUDA.
The thing that’s not as fast is the backward, so writing your own scripted operation there might be an option - I’m guessing you don’t want to wait until the better backward fusing is merged.
Best regards
Thomas