[❓질문] pytorch 이런 식은 보통 어떻게 구현하나요?

[1(222.100)]

q_0 = l_0이고 i>=1 일때 재귀적으로 구해가는 식입니다.

이러면 for문을 돌려야 하는데 그러면 Forward도 Backward도 속도가 조금 구리지 않나요?

일단 기능 구현은 했는데 jit 같은걸 써야 할까요?

어디보면

@cuda.jit / cuda.syncthreads() 써서 Backward 함수까지 구현해놓은 곳도 있던데 이런것보다 간편하게 고속화 하는 방법이 있을까요?