bamsolve

Predict photonic circuits with AI. The API is free during the public alpha.

Get started

1. Install the package:

python -m pip install --upgrade bamsolve

2. Enter your email at https://bamsolve.com/#get-key. Copy the API key shown on the page.
3. Run the example below. It downloads a ready-to-use GDS and asks for your key with hidden input.

from getpass import getpass
from urllib.request import urlretrieve
from bamsolve import Layer, Stack, bamsolve, pred

urlretrieve(
    "https://bamsolve.com/examples/ebeam_y_1550.gds",
    "ebeam_y_1550.gds",
)
bamsolve(api_key=getpass("Bamsolve API key: "))

stack = Stack(clad=1.444, layers={(1, 0): Layer(n=3.48, z=(0.0, 0.22))})
dev = stack("ebeam_y_1550.gds")
res = pred(dev, inputs={((1, 10), "TE", 1.55)})

print("S shape:", res.S.shape)  # (1, 3, 3)
print("Upper-arm power:", abs(res.S[0, 1, 0]) ** 2)
print("Lower-arm power:", abs(res.S[0, 2, 0]) ** 2)
res.export_s("ebeam_y_1550.s3p")

The example is a 15 µm silicon Y-branch with three ports. Its core is on GDS layer/datatype (1, 0). Port opt1 is at the left end on (1, 10). Port opt2 is at the upper-right end on (1, 11). Port opt3 is at the lower-right end on (1, 12). The z bounds and wavelength are in micrometres. Refractive indices have no units.

This device includes a split and two curved arms. A full 3D FDTD test must resolve these features, the core thickness, and the surrounding cladding. A direct full-matrix calculation also needs a separate input at each port. This takes more work than a straight-waveguide check. The tutorial does not give an FDTD runtime for your computer; runtime depends on the mesh, simulation duration, and hardware.

res.S contains every input and output port for the returned polarization, with shape (wavelength, output_port, input_port). In this example, array index 0 is the left port, index 1 is the upper-right port, and index 2 is the lower-right port. The input tuple selects the field shown in res.E; it does not restrict the S-matrix to one column. res.n_eff holds the effective index. res.E is a complex array indexed as [y, x], on the coordinates res.y and res.x in micrometres.

Read the Get started tutorial for downloadable scripts, a simple straight-waveguide check, port instructions, and plotting. You can also create the straight GDS with gdstk. The downloaded GDS does not require gdstk to run a prediction.

Check the exported file

Install an independent Touchstone reader with python -m pip install scikit-rf, then run:

import numpy as np
import skrf

network = skrf.Network("ebeam_y_1550.s3p")
assert np.allclose(network.s, res.S, atol=1e-7)
print(network.f)  # Frequency in Hz

This checks file compatibility. It does not independently validate the predicted physics.

API configuration

The default API address is https://api.bamsolve.com.

You can set BAMSOLVE_API_KEY before starting Python instead of entering the key in your script. Keep the key out of source control and shared notebooks.

To use a custom server, set its URL before you call pred:

bamsolve(api_key="bsk_...", url="https://YOUR_API_HOST")

You can also set the BAMSOLVE_URL environment variable before you import the module.