Stars
Electromagnetic simulation (RCWA) & optimization package in Python
Efficient Differentiable n-d PDE solvers in JAX.
Prototype for sparse automatic differentiation in JAX + Julia
Surrogate modeling and optimization for scientific machine learning (SciML)
GPU-accelerated RCWA with automatic differentiation
Fourier Modal Method with Jax (official maintained repo)
Database of optical constants
A fast transfer matrix method written in jax for modelling optical multilayer thin films
Rigorous Coupled Wave Analyses (RCWA) for layered structures with incommensurate periodicities
Optax is a gradient processing and optimization library for JAX.
Artificial nanofabrication of integrated photonic circuits using deep learning
High-performance quantum systems simulation with JAX (GPU-accelerated & differentiable solvers).
Python library for making 3D plots with blender
jax-triton contains integrations between JAX and OpenAI Triton
Gdstk (GDSII Tool Kit) is a C++/Python library for creation and manipulation of GDSII and OASIS files.
An opensource python implementation of the Method of moving asymptotes (MMA)
A GPU-accelerated, differentiable, Maxwell FDTD solver
Topology Optimization of Photonic Components using a Shape Library
Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
A Python framework for accelerated simulation, data generation and spatial computing.
🌱 Guided-mode expansion of photonic crystal slabs
A massively parallel, high-level programming language
TORAX: Tokamak transport simulation in JAX
A JAX research toolkit for building, editing, and visualizing neural networks.