published Published 10 months ago
jaxhps: An elliptic PDE solver built with machine learning in mind
Python
published Published 11 months ago
TMMax: High-performance modeling of multilayer thin-film structures using transfer matrix method with JAX
Python Jupyter Notebook
published Published 11 months ago
JaxLayerLumos: A JAX-based Differentiable Optical and Radio Frequency Simulator for Multilayer Structures
Python
published Published 12 months ago
Graphem - JAX: Node Influence Maximization via Geometric Embeddings
Python
published Published about 1 year ago
ginjax: E(d)-Equivariant CNN for Tensor Images
Jupyter Notebook Python
published Published about 1 year ago

