TorchOptics Documentation#
TorchOptics is an open-source Python library for simulating optical systems using Fourier optics, built on PyTorch. It provides GPU-accelerated, fully differentiable wave optics simulations, enabling end-to-end optimization of optical hardware jointly with machine learning models.
Key Features#
Model, analyze, and optimize optical systems using Fourier optics.
GPU acceleration, batch processing, and automatic differentiation.
Joint optimization of optical hardware and machine learning models.
Lenses, phase/amplitude modulators, detectors, polarizers, and more.
Hermite-Gaussian, Laguerre-Gaussian, Zernike modes, gratings, and others.
Simulate polarized light and fields with arbitrary spatial coherence.
Installation#
pip install torchoptics
Contributing#
Contributions are welcome! See the Contributing Guide for details.
Citing TorchOptics#
If you use TorchOptics in your research, please cite our paper.
License#
Distributed under the MIT License. See LICENSE for details.