Updated
CUDA
Links
https://github.com/HMUNACHI/henry-vjp