Metadata-Version: 2.4
Name: beamz
Version: 0.5.0
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Operating System :: OS Independent
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Classifier: Topic :: Scientific/Engineering :: Physics
Requires-Dist: numpy>=1.24.4
Requires-Dist: scipy>=1.10.1
Requires-Dist: shapely>=2.0.6
Requires-Dist: jax>=0.4.0
Requires-Dist: optax>=0.1.0
Requires-Dist: matplotlib>=3.8.0
Requires-Dist: scikit-image>=0.21.0
Requires-Dist: xarray>=2024.0.0
Requires-Dist: ipykernel>=6.29.0 ; extra == 'dev'
Requires-Dist: maturin>=1.9,<2 ; extra == 'dev'
Requires-Dist: meshio>=5.3 ; extra == 'dev'
Requires-Dist: nbconvert>=7.16.0 ; extra == 'dev'
Requires-Dist: trimesh>=4.5 ; extra == 'dev'
Requires-Dist: pytest>=7.0.0 ; extra == 'dev'
Requires-Dist: pytest-cov>=4.0.0 ; extra == 'dev'
Requires-Dist: myst-parser>=2.0.0 ; extra == 'dev'
Requires-Dist: mkdocs>=1.5.0 ; extra == 'dev'
Requires-Dist: mkdocs-material>=9.0.0 ; extra == 'dev'
Requires-Dist: pyright>=1.1.400 ; extra == 'dev'
Requires-Dist: ruff>=0.14.0 ; extra == 'dev'
Requires-Dist: vulture>=2.14 ; extra == 'dev'
Requires-Dist: gdsfactory>=8.0 ; python_full_version < '3.14' and extra == 'gds'
Requires-Dist: gdsfactory>=9.0 ; python_full_version >= '3.14' and extra == 'gds'
Requires-Dist: pyright>=1.1.400 ; extra == 'lint'
Requires-Dist: ruff>=0.14.0 ; extra == 'lint'
Requires-Dist: vulture>=2.14 ; extra == 'lint'
Requires-Dist: meshio>=5.3 ; extra == 'mesh'
Requires-Dist: trimesh>=4.5 ; extra == 'mesh'
Requires-Dist: pytest>=7.0.0 ; extra == 'test'
Requires-Dist: pytest-cov>=4.0.0 ; extra == 'test'
Requires-Dist: pytest-randomly>=3.0.0 ; extra == 'test'
Requires-Dist: hypothesis>=6.100.0 ; extra == 'test'
Requires-Dist: diff-cover>=9.0.0 ; extra == 'test'
Requires-Dist: gdsfactory>=8.0 ; python_full_version < '3.14' and extra == 'ubc'
Requires-Dist: gdsfactory>=9.0 ; python_full_version >= '3.14' and extra == 'ubc'
Requires-Dist: ubcpdk==2.7.0 ; python_full_version < '3.14' and extra == 'ubc'
Provides-Extra: dev
Provides-Extra: gds
Provides-Extra: lint
Provides-Extra: mesh
Provides-Extra: test
Provides-Extra: ubc
License-File: LICENSE
Summary: Open GPU-accelerated FDTD for photonic chip designers.
Keywords: computational physics,electromagnetics,photonics,inverse design,simulation
Author-email: Quentin Wach <quentin.wach+beamz@gmail.com>
Requires-Python: >=3.10, <3.15
Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
Project-URL: Bug Tracker, https://github.com/beamzorg/beamz/issues
Project-URL: Documentation, https://www.beamz.tech/docs/index
Project-URL: Homepage, https://github.com/beamzorg/beamz
Project-URL: Repository, https://github.com/beamzorg/beamz

<div align="center">
  <picture style="padding-right: 0px; padding-bottom: 7px;">
    <source media="(prefers-color-scheme: light)" srcset="docs/assets/beamz_logo_black.png">
    <img alt="BeamZ logo" src="docs/assets/beamz_logo_white.png" width="130">
  </picture>

  <strong>BeamZ</strong> is a <strong>GPU-accelerated</strong> <strong><a href="https://en.wikipedia.org/wiki/Electromagnetism">electromagnetic</a> simulation</strong> framework for photonic chip designers using the <strong><a href="https://en.wikipedia.org/wiki/Finite-difference_time-domain_method">FDTD</a> method</strong>. It enables fast, large-scale simulations and offers a <strong>familiar, high-level API</strong> for fast prototyping with just a few lines of code as well as an <strong>inverse design module</strong> for gradient-based optimization using the <strong>adjoint method</strong>.

  <h3>

  [Homepage](https://www.beamz.tech) / [Documentation](https://www.beamz.tech/docs/index) / [Example Library](https://www.beamz.tech/examples)

  </h3>

  [![License: Apache-2.0](https://img.shields.io/badge/License-Apache--2.0-blue.svg)](https://github.com/beamzorg/beamz/blob/main/LICENSE)
  [![Tests](https://github.com/beamzorg/beamz/actions/workflows/tests.yml/badge.svg?branch=main)](https://github.com/beamzorg/beamz/actions/workflows/tests.yml)
  [![Coverage](https://raw.githubusercontent.com/beamzorg/beamz/main/.github/badges/coverage.svg)](https://github.com/beamzorg/beamz/actions/workflows/tests.yml)
</div>



## Core Features
- **Python-first**, free (Apache-2.0 license) & open-source, with a native Rust rasterizer.
- FDTD simulation in **2D and 3D**.
- **GPU-accelerated**, achieving high **GCUPS performance**.
- **Multi-GPU** runs, handling **large-scale simulations** with _billions of cells_.
- CPU-capable for **fast prototyping**, even on your laptop.
- Intuitive and **familiar API**.
- Native **FDFD mode solver** with discrete Yee-grid refinement and validation.
- **CPML**, absorbing layers and PEC boundaries.
- Unidirectional **mode sources** (single freq. and broadband, Huygens fields + TFSF, TE/TM).
- **Gaussian sources**, e.g. for grating coupler simulations.
- Integrated **rasterization module**.
- **Sub-pixel averaging** using super-sampling.
- Custom source time profiles.
- Built-in layout flow (GDSII import/export).
- **DFT monitors** and S-parameter extraction workflow for compact modeling.
- Streamlined **parametric design** module.
- Optimization/autodiff utilities for gradient-based **inverse-design** with Jax.


## Examples
Try out notebooks from our growing **[example library](https://beamz.tech/examples/)**. It includes:

- [1) Mode Sources and Monitors](https://beamz.tech/examples/modal_sources_monitors)
- [2) Waveguide Crossing with Cosine Tapers](https://beamz.tech/examples/cosine_waveguide_crossing)
- [3) Topology Optimized 90° Bend (2D)](https://beamz.tech/examples/ceviche_bend)
<!--- [Broadband Mode Sources]() (coming soon)
- [Straight & Curved Waveguide Benchmark]() (coming soon)-->
<!--- [Mode Converter (3D)]() (coming soon)
- [DEMUX]() (coming soon)-->

## Integration 

BeamZ is used by several other OSS packages as an FDTD engine:
+ [SiEPIC's GDS FDTD](https://github.com/SiEPIC/gds_fdtd), an EDA- and solver-agnostic 3D FDTD compact modeling framework.
+ [Lumix](https://github.com/amiskandarmuda/lumix), a research codebase for optical neural networks and matrix inverse design.


## Installation

Install BeamZ using pip:

```bash
pip install beamz
```

Development uses [uv](https://docs.astral.sh/uv/). Clone the repository and sync
the package with its contributor dependencies:

```bash
git clone https://github.com/beamzorg/beamz
cd beamz
uv sync --extra dev --extra test
```

For a ready-to-use CUDA and Jupyter development environment, see the
[Docker and RunPod guide](docker/runpod/README.md).


## About
BeamZ's mission is to be the **pragmatic** FDTD engine of choice for **photonic chip designers**.

It focuses on **streamlined workflows** over **feature bloat** to produce **useful results** without tedious setup or configuration files and bringing GPU-acceleration for **maximum performance in large-scale simulations** to everyone.

The project is **actively maintained**. We aim to keep the code in Python, minimize dependencies, keep the line-count low, commented, and features local within the code to **make the code readable and development easy** so that - if there is something that isn't working or missing - you can quickly add it yourself. The engine is grounded in hundreds of tests, verifiable simulations and benchmarks, replicating known results from the established literature. Beyond benchmarking the core engine stats, we aim to **reduce friction for chip designers at every step** - from installation, to setting up the sim using a familiar API, to optimizing the performance of the rasterizer, mode solver, compiler, optimization loop, and integration into the overall chip design workflow.


## Contributing

**We appreciate all contributions.** If you are planning to contribute bug-fixes, please do so without any further discussion. If you would like to add new features, please first open an issue and discuss the feature with us. There may be ongoing work that could conflict with your changes, or we may be heading in a different direction and we don't want to waste your time working on something that might be rejected. - You can find [more information here](CONTRIBUTING.md).

The simplest way to support the project of course is by **giving this repo a star.** Thank you!

---

Copyright © 2026 Quentin Wach — [Apache-2.0](https://github.com/beamzorg/beamz/blob/HEAD/LICENSE)

