bioRxiv · 10.64898/2026.03.10.710885
torch-projectors: A High-Performance Differentiable Projection Library for PyTorch
Abstract
Fourier-space projection operations are central to electron microscopy single-particle analysis and electron tomography algorithms. Machine learning methods require differentiable implementations for end-to-end model training, but PyTorchs built-in operations are too slow for practical use. This paper introduces torch-projectors: a high-performance library for differentiable Fourier-space projections in PyTorch. The library provides 2D and 3D forward and backward projection operators with linear and cubic interpolation, supporting gradient calculation for all inputs. Optimized for CPU, Apple Silicon (MPS), and CUDA devices, torch-projectors outperforms torch-fourier-slice by 1-2 orders of magnitude.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Tegunov, D.. 2026-03-10. torch-projectors: A High-Performance Differentiable Projection Library for PyTorch. https://doi.org/10.64898/2026.03.10.710885
Cite the original work for its findings. Save a collection to share your selection of sources.