tsne-cuda

A optimized CUDA version of FIt-SNE algorithm with associated python modules. We find that our implementation of t-SNE can be up to 1200x faster than Sklearn, or up to 50x faster than Multicore-TSNE when used with the right GPU. The paper describing our approach, as well as the results below, is available at https://arxiv.org/abs/1807.11824.

More Information

For more information visit the tsne-cuda website.

Available Versions

These versions of tsne-cuda are available on Baskerville. These will be retained in accordance with our Applications Support and Retention Policy.