Ceyron/DeepONet-in-JAX
Simple implementation of Deep Operator Networks (DeepONets) in the JAX deep learning framework together with Equinox.
Jupyter NotebookMIT
Simple implementation of Deep Operator Networks (DeepONets) in the JAX deep learning framework together with Equinox.
Jupyter NotebookMIT