TorchGDM: A GPU-Accelerated Python Toolkit for Multi-Scale Electromagnetic Scattering with Automatic Differentiation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ponomareva, Sofia, Patoux, Adelin, Majorel, Clément, Azéma, Antoine, Cuche, Aurélien, Girard, Christian, Arbouet, Arnaud, Wiecha, Peter R.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918164912144384
author Ponomareva, Sofia
Patoux, Adelin
Majorel, Clément
Azéma, Antoine
Cuche, Aurélien
Girard, Christian
Arbouet, Arnaud
Wiecha, Peter R.
author_facet Ponomareva, Sofia
Patoux, Adelin
Majorel, Clément
Azéma, Antoine
Cuche, Aurélien
Girard, Christian
Arbouet, Arnaud
Wiecha, Peter R.
contents We present "torchGDM", a numerical framework for nano-optical simulations based on the Green's Dyadic Method (GDM). This toolkit combines a hybrid approach, allowing for both fully discretized nano-structures and structures approximated by sets of effective electric and magnetic dipoles. It supports simulations in three dimensions and for infinitely long, two-dimensional structures. This capability is particularly suited for multi-scale modeling, enabling accurate near-field calculations within or around a discretized structure embedded in a complex environment of scatterers represented by effective models. Importantly, torchGDM is entirely implemented in PyTorch, a well-optimized and GPU-enabled automatic differentiation framework. This allows for the efficient calculation of exact derivatives of any simulated observable with respect to various inputs, including positions, wavelengths or permittivity, but also intermediate parameters like Green's tensor components, which can be interesting for physics informed deep learning applications. We anticipate that this toolkit will be valuable for applications merging nano-photonics and machine learning, as well as for solving nano-photonic optimization and inverse problems, such as the global design and characterization of metasurfaces, where optical interactions between structures are critical.
format Preprint
id arxiv_https___arxiv_org_abs_2505_09545
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TorchGDM: A GPU-Accelerated Python Toolkit for Multi-Scale Electromagnetic Scattering with Automatic Differentiation
Ponomareva, Sofia
Patoux, Adelin
Majorel, Clément
Azéma, Antoine
Cuche, Aurélien
Girard, Christian
Arbouet, Arnaud
Wiecha, Peter R.
Optics
Computational Physics
We present "torchGDM", a numerical framework for nano-optical simulations based on the Green's Dyadic Method (GDM). This toolkit combines a hybrid approach, allowing for both fully discretized nano-structures and structures approximated by sets of effective electric and magnetic dipoles. It supports simulations in three dimensions and for infinitely long, two-dimensional structures. This capability is particularly suited for multi-scale modeling, enabling accurate near-field calculations within or around a discretized structure embedded in a complex environment of scatterers represented by effective models. Importantly, torchGDM is entirely implemented in PyTorch, a well-optimized and GPU-enabled automatic differentiation framework. This allows for the efficient calculation of exact derivatives of any simulated observable with respect to various inputs, including positions, wavelengths or permittivity, but also intermediate parameters like Green's tensor components, which can be interesting for physics informed deep learning applications. We anticipate that this toolkit will be valuable for applications merging nano-photonics and machine learning, as well as for solving nano-photonic optimization and inverse problems, such as the global design and characterization of metasurfaces, where optical interactions between structures are critical.
title TorchGDM: A GPU-Accelerated Python Toolkit for Multi-Scale Electromagnetic Scattering with Automatic Differentiation
topic Optics
Computational Physics
url https://arxiv.org/abs/2505.09545