Kernel Operations on the GPU, with Autodiff, without Memory Overflows

Benjamin Charlier, Jean Feydy, Joan Alexis Glaunès, François-David Collin, Ghislain Durif. Kernel Operations on the GPU, with Autodiff, without Memory Overflows. Journal of Machine Learning Research, 22, 2021. [doi]

@article{CharlierFGCD21,
  title = {Kernel Operations on the GPU, with Autodiff, without Memory Overflows},
  author = {Benjamin Charlier and Jean Feydy and Joan Alexis Glaunès and François-David Collin and Ghislain Durif},
  year = {2021},
  url = {http://jmlr.org/papers/v22/20-275.html},
  researchr = {https://researchr.org/publication/CharlierFGCD21},
  cites = {0},
  citedby = {0},
  journal = {Journal of Machine Learning Research},
  volume = {22},
}