Abstract
Unsupervised machine learning has recently gained much attention in the field of molecular dynamics (MD). Particularly, dimensionality reduction techniques have been regularly employed to analyze large volumes of high-dimensional MD data to gain insight into hidden information encoded in MD trajectories. Among many such techniques, t-distributed stochastic neighbor embedding (t-SNE) is especially popular. A parametric version of t-SNE that employs neural networks is less commonly known, yet it has demonstrated superior performance in dimensionality reduction compared to the standard implementation. Here, we present a Python package called NeuralTSNE with our implementation of parametric t-SNE. The implementation is done using the PyTorch library and the PyTorch Lightning framework and can be imported as a module or used from the command line. We show that NeuralTSNE offers an easy-to-use tool for the analysis of MD data.