CannyLab / tsne-cuda

GPU Accelerated t-SNE for CUDA with Python bindings
BSD 3-Clause "New" or "Revised" License
1.78k stars 126 forks source link

Gradient problem #125

Closed fengyuxie closed 1 year ago

fengyuxie commented 1 year ago

This project has been implemented very well. It would be great if we could replace the numpy in it to support gradients. The current version of the code cannot be directly used in neural networks (as there will be errors if there are no gradients).

DavidMChan commented 1 year ago

We're happy that you appreciate the project, however it's unlikely that this code will ever support gradients, as tsne is a non-parametric algorithm, and the computation required to support gradients would not only be a significant mathematical undertaking, but add significant overhead to the forward pass.