jaxlayerlumos

JaxLayerLumos: A JAX-based Differentiable Optical and Radio Frequency Simulator for Multilayer Structures

This package provides tools for calculating reflection and transmission coefficients for multilayer optical stacks using JAX for efficient computation and automatic differentiation. It supports both TE and TM polarizations, various materials, and different light sources.

Main Functions:

• stackrt_n_k: Calculate reflection/transmission for refractive index stacks

• stackrt_eps_mu: Calculate reflection/transmission for permittivity/permeability stacks

• stackrt_n_k_0: Wrapper for normal incidence (0 degrees)

• stackrt_n_k_45: Wrapper for 45-degree incidence

The package includes utilities for:

• Material properties and conversions

• Light source spectra (AM0, AM1.5G, AM1.5D)

• Color calculations and transformations

• Unit conversions

• Position and geometry calculations

• Radio frequency and radar materials

• jaxlayerlumos.jaxlayerlumos
  • false_fun()
  • stackrt_eps_mu()
  • stackrt_eps_mu_base()
  • stackrt_eps_mu_theta()
  • stackrt_n_k()
  • true_fun()
• jaxlayerlumos.utils_layers
  • get_thicknesses_surrounded_by_air()
• jaxlayerlumos.utils_light_sources
  • get_irradiance()
  • interpolate()
  • interpolate_light_source_irradiance()
  • load_light_source_json()
  • load_light_source_wavelength()
  • load_light_source_wavelength_nm()
• jaxlayerlumos.utils_materials
  • convert_n_k_to_eps_mu_for_non_magnetic_materials()
  • get_all_materials()
  • get_eps_mu()
  • get_n_k()
  • get_n_k_surrounded_by_air()
  • interpolate()
  • interpolate_material_eps_mu()
  • interpolate_material_n_k()
  • load_json()
  • load_material()
  • load_material_wavelength()
  • load_material_wavelength_um()
• jaxlayerlumos.utils_position
  • calc_absorption_in_each_layer()
  • calc_position_data()
  • calc_position_in_structure()
• jaxlayerlumos.utils_radar_materials
  • get_eps_mu_Michielssen()
• jaxlayerlumos.utils_radio_frequency
  • load_material_RF()
• jaxlayerlumos.utils_spectra
  • convert_frequencies_to_wavelengths()
  • convert_wavelengths_to_energy()
  • convert_wavelengths_to_frequencies()
  • get_frequencies_visible_light()
  • get_frequencies_wide_visible_light()
• jaxlayerlumos.utils_units
  • convert_cm_to_m()
  • convert_m_to_nm()
  • convert_m_to_um()
  • convert_mm_to_m()
  • convert_nm_to_m()
  • get_centi()
  • get_elementary_charge()
  • get_giga()
  • get_light_speed()
  • get_micro()
  • get_milli()
  • get_nano()
  • get_planck_constant()
• jaxlayerlumos.wrappers
  • stackrt_n_k_0()
  • stackrt_n_k_45()
• jaxlayerlumos.colors
  • jaxlayerlumos.colors.color_matching_functions
  • jaxlayerlumos.colors.composite
    • spectrum_to_Lab()
    • spectrum_to_sRGB()
    • spectrum_to_xyY()
  • jaxlayerlumos.colors.illuminants
  • jaxlayerlumos.colors.transform
    • XYZ_to_Lab()
    • XYZ_to_sRGB()
    • XYZ_to_xyY()
    • spectrum_to_XYZ()
  • jaxlayerlumos.colors.utils
    • get_cmfs()
    • get_illuminant()