TorchOptics: An open-source Python library for differentiable Fourier optics simulations

Bibliographic Details
Title: TorchOptics: An open-source Python library for differentiable Fourier optics simulations
Authors: Filipovich, Matthew J., Lvovsky, A. I.
Publication Year: 2024
Collection: Physics (Other)
Subject Terms: Physics - Optics, Physics - Classical Physics, Physics - Computational Physics
Description: TorchOptics is an open-source Python library for differentiable Fourier optics simulations, developed using PyTorch to enable GPU-accelerated tensor computations and automatic differentiation. It provides a comprehensive framework for modeling, analyzing, and designing optical systems using Fourier optics, with applications in imaging, diffraction, holography, and signal processing. The library leverages PyTorch's automatic differentiation engine for gradient-based optimization, enabling the inverse design of complex optical systems. TorchOptics supports end-to-end optimization of hybrid models that integrate optical systems with machine learning architectures for digital post-processing. The library includes a wide range of optical elements and spatial profiles, and supports simulations with polarized light and fields with arbitrary spatial coherence.
Comment: 9 pages, 6 figures
Document Type: Working Paper
DOI: 10.48550/arXiv.2411.18591
Access URL: http://arxiv.org/abs/2411.18591
Accession Number: edsarx.2411.18591
Database: arXiv
Description
DOI:10.48550/arXiv.2411.18591