/DeepONet-in-JAX

Simple implementation of Deep Operator Networks (DeepONets) in the JAX deep learning framework together with Equinox.

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers