Skip to content

General

Generalยค

This page contains the API reference for the building blocks of Neural Operators in neojax.

The pre-built models (FNO, TFNO, ...) are built from these components. They can also be composed directly, using standard equinox composition, to construct custom neural operator architectures.